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

Refactoring detection task #3860

Merged
merged 28 commits into from
Aug 27, 2024
Merged
Changes from 1 commit
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
4d421d1
Rename directory from `base_models` to `detectors`
sungchul2 Aug 16, 2024
e1d156d
Update pytorchcv version
sungchul2 Aug 16, 2024
7a3b01d
Create criterion modules
sungchul2 Aug 16, 2024
ef0c1c2
Fix unit tests
sungchul2 Aug 21, 2024
b0abc18
Update comment
sungchul2 Aug 21, 2024
32b5a2a
Merge branch 'develop' into refactoring-detection
sungchul2 Aug 21, 2024
a13ad86
FIx iseg unit test
sungchul2 Aug 22, 2024
8098edd
Merge branch 'develop' into refactoring-detection
sungchul2 Aug 22, 2024
9cf1e24
Enable factory for ATSS
sungchul2 Aug 22, 2024
b4ee9f1
Enable factory for YOLOX and update
sungchul2 Aug 22, 2024
2d595df
Enable factory for RTMDet
sungchul2 Aug 22, 2024
93f3f5c
Update recipes
sungchul2 Aug 22, 2024
d56b1ce
Update
sungchul2 Aug 22, 2024
055fe84
Enable factory for SSD
sungchul2 Aug 22, 2024
3d7187b
Fix unit tests
sungchul2 Aug 22, 2024
bbb8116
Enable factory for RTDETR
sungchul2 Aug 23, 2024
6428882
Reduce default parameters
sungchul2 Aug 23, 2024
f469b90
Update huggingface, keypoint, and iseg
sungchul2 Aug 23, 2024
1846d0a
Revert default `input_size` argument
sungchul2 Aug 23, 2024
0a2471e
Fix unit test
sungchul2 Aug 23, 2024
cf6485a
Fix integration test
sungchul2 Aug 23, 2024
c3dfef0
Add ABC
sungchul2 Aug 23, 2024
4277eba
Fix
sungchul2 Aug 23, 2024
49f1383
Merge branch 'develop' into refactoring-detection
sungchul2 Aug 23, 2024
6287f82
Change `model_version` to `model_name`
sungchul2 Aug 23, 2024
68abb0a
precommit
sungchul2 Aug 23, 2024
f38b613
Remove `DetectionBackboneFactory`
sungchul2 Aug 26, 2024
c3f5668
Use `Literal` for `model_name`
sungchul2 Aug 26, 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
precommit
  • Loading branch information
sungchul2 committed Aug 23, 2024
commit 68abb0ac86f74920d991b2f276a768f96b820b65
6 changes: 3 additions & 3 deletions src/otx/algo/detection/atss.py
Original file line number Diff line number Diff line change
@@ -39,9 +39,9 @@
"atss_resnext101",
]

PRETRAINED_ROOT: str = (
"https://storage.openvinotoolkit.org/repositories/openvino_training_extensions/models/object_detection/v2/"
)
PRETRAINED_ROOT: (
str
) = "https://storage.openvinotoolkit.org/repositories/openvino_training_extensions/models/object_detection/v2/"

PRETRAINED_WEIGHTS: dict[str, str] = {
"atss_mobilenetv2": PRETRAINED_ROOT + "mobilenet_v2-atss.pth",
6 changes: 3 additions & 3 deletions src/otx/algo/detection/rtmdet.py
Original file line number Diff line number Diff line change
@@ -35,9 +35,9 @@

AVAILABLE_MODEL_VERSIONS: list[str] = ["rtmdet_tiny"]

PRETRAINED_ROOT: str = (
"https://storage.openvinotoolkit.org/repositories/openvino_training_extensions/models/object_detection/v2/"
)
PRETRAINED_ROOT: (
str
) = "https://storage.openvinotoolkit.org/repositories/openvino_training_extensions/models/object_detection/v2/"

PRETRAINED_WEIGHTS: dict[str, str] = {
"rtmdet_tiny": PRETRAINED_ROOT + "rtmdet_tiny.pth",
6 changes: 3 additions & 3 deletions src/otx/algo/detection/ssd.py
Original file line number Diff line number Diff line change
@@ -45,9 +45,9 @@

AVAILABLE_MODEL_VERSIONS: list[str] = ["ssd_mobilenetv2"]

PRETRAINED_ROOT: str = (
"https://storage.openvinotoolkit.org/repositories/openvino_training_extensions/models/object_detection/v2/"
)
PRETRAINED_ROOT: (
str
) = "https://storage.openvinotoolkit.org/repositories/openvino_training_extensions/models/object_detection/v2/"

PRETRAINED_WEIGHTS: dict[str, str] = {
"ssd_mobilenetv2": PRETRAINED_ROOT + "mobilenet_v2-2s_ssd-992x736.pth",
Loading