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

Update visual prompting task (finetuning & zero-shot learning) #2892

Merged
merged 30 commits into from
Feb 23, 2024
Merged
Changes from 1 commit
Commits
Show all changes
30 commits
Select commit Hold shift + click to select a range
acdef2f
Add handling empty predicted tensor
sungchul2 Feb 6, 2024
d6978d5
Update image encoder
sungchul2 Feb 6, 2024
61d8596
Enable inner postprocessing
sungchul2 Feb 6, 2024
b4d66b2
Update OV inferencer (TODO: zsl)
sungchul2 Feb 8, 2024
205c763
Update `learn` pipeline
sungchul2 Feb 13, 2024
e661f0a
Update `infer` pipeline
sungchul2 Feb 13, 2024
a6add5d
[WIP] Update for export
sungchul2 Feb 13, 2024
df28d1a
Update data & forward pipeline for multiple prompts
sungchul2 Feb 15, 2024
b8d89e4
(WIP) Update zsl pipeline
sungchul2 Feb 15, 2024
bdd7fc5
Update OV inference
sungchul2 Feb 16, 2024
358d97b
Fix for unit & intg tests
sungchul2 Feb 16, 2024
cc60bb7
precommit
sungchul2 Feb 16, 2024
4eb9312
Fix e2e
sungchul2 Feb 19, 2024
dc059ab
precommit
sungchul2 Feb 19, 2024
a6e460d
Fix unit tests
sungchul2 Feb 19, 2024
387edff
Revert postprocessing
sungchul2 Feb 19, 2024
182255d
precommit
sungchul2 Feb 19, 2024
5637852
Merge branch 'develop' into update_vpm
sungchul2 Feb 19, 2024
e171792
Update unit tests
sungchul2 Feb 20, 2024
43bc6c2
precommit
sungchul2 Feb 20, 2024
3966f36
Fix
sungchul2 Feb 20, 2024
b3448f6
Fix for unit test coverage
sungchul2 Feb 20, 2024
5c6ffd2
Refactor model loading and state dict handling
sungchul2 Feb 22, 2024
539a37c
Remove unnecessary flag
sungchul2 Feb 22, 2024
a01345a
precommit
sungchul2 Feb 22, 2024
d030427
Fix
sungchul2 Feb 22, 2024
9d144fd
Fix
sungchul2 Feb 22, 2024
a99d3b9
Fix
sungchul2 Feb 22, 2024
d5df1b9
Merge branch 'develop' into update_vpm
sungchul2 Feb 23, 2024
826408f
Fix unittest
sungchul2 Feb 23, 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
Prev Previous commit
Next Next commit
Remove unnecessary flag
sungchul2 committed Feb 22, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
commit 539a37cbd3ec058d70e86704ba159d7aa9f53d26
4 changes: 2 additions & 2 deletions src/otx/algorithms/visual_prompting/tasks/openvino.py
Original file line number Diff line number Diff line change
@@ -491,15 +491,15 @@ def forward_decoder( # type: ignore
)
has_mask_input = self.has_mask_inputs[0]

elif is_cascade and i == 1:
elif i == 1:
# Cascaded Post-refinement-1
mask_input, masks = self._postprocess_masks(masks, logits, scores, is_single=True) # noqa: F821
if masks.sum() == 0:
return {"upscaled_masks": masks}

has_mask_input = self.has_mask_inputs[1]

elif is_cascade and i == 2:
elif i == 2:
# Cascaded Post-refinement-2
mask_input, masks = self._postprocess_masks(masks, logits, scores) # noqa: F821
if masks.sum() == 0: