Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Multi-head model support #149 #150

Merged
merged 31 commits into from
Mar 21, 2024
Merged

Conversation

przemyslaw-aszkowski
Copy link
Contributor

@przemyslaw-aszkowski przemyslaw-aszkowski commented Feb 21, 2024

TODO:

  • Remove the clip for the segmentation model in the postprocessing function
  • Allow for multi outputs for regressor and segmentor
  • Fix get_channel_name due to changes
  • Fix _create_result_message due to changes
  • Update docs
  • Add an example for a multi-head model for segmentation
  • Add an example for a two-class model for object detection
  • In-depth evaluation

@bartoszptak bartoszptak changed the title Clip removed and docs updated #149 Multi-head model support #149 Feb 23, 2024
result_imgs.append(result_img)

return result_imgs
return self.limit_extended_extent_image_to_base_extent_with_mask(full_img=full_imgs)
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

maybe rename parameter name to full_imgs for consistency

@@ -1,6 +1,8 @@
""" This file implements map processing for segmentation model """

from calendar import c
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

random import

@bartoszptak
Copy link
Contributor

bartoszptak commented Mar 20, 2024

@przemyslaw-aszkowski I think the PR is ready for review and merge.

Edit: Nope, there is something wrong with export features.

@bartoszptak
Copy link
Contributor

Now works fine!

@przemyslaw-aszkowski przemyslaw-aszkowski merged commit 54c5868 into devel Mar 21, 2024
2 checks passed
@bartoszptak bartoszptak deleted the clip_removed_and_docs_update branch March 21, 2024 11:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants