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.

Sign up for GitHub

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

Update has_mask method for mmdet models #1054

Merged
merged 11 commits into from
Jun 7, 2024
Merged

Update has_mask method for mmdet models #1054

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
cf0a3e8
Fix TypeError: 'GeometryCollection' object is not subscriptable when …
Alias-z May 31, 2024
0afbf5b
Test lint fixing with isort and black
Alias-z Jun 2, 2024
893fe7b
Merge branch 'main' into main
Alias-z Jun 2, 2024
fad86cf
Attempt with run_code_style format
Alias-z Jun 2, 2024
adedfd2
Merge branch 'obss:main' into main
Alias-z Jun 2, 2024
edc260a
Update has_mask method for mmdet models
Alias-z Jun 4, 2024
6965f84
Merge branch 'main' of https://github.com/Alias-z/sahi
Alias-z Jun 4, 2024
68ad269
update mmdet test configs
Alias-z Jun 4, 2024
0156fb9
Revert "update mmdet test configs"
Alias-z Jun 4, 2024
9c8aff5
update has_mask method
Alias-z Jun 4, 2024
9f26c66
Reformat with black and isort
Alias-z Jun 4, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion sahi/models/mmdet.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,9 @@ def has_mask(self):
"""
Returns if model output contains segmentation mask
"""
has_mask = self.model.model.with_mask
# has_mask = self.model.model.with_mask
train_pipeline = self.model.cfg["train_dataloader"]["dataset"]["pipeline"]
has_mask = any(isinstance(item, dict) and any("mask" in key for key in item.keys()) for item in train_pipeline)
return has_mask

@property
Expand Down