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

OTX D-Fine Detection Algorithm Integration #4142

Open
wants to merge 51 commits into
base: develop
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
51 commits
Select commit Hold shift + click to select a range
159c2fc
init
eugene123tw Nov 26, 2024
68b6a0d
remove convertbox
eugene123tw Nov 26, 2024
7a35df4
Refactor D-FINE detector: remove unused components and update model c…
eugene123tw Nov 26, 2024
c0d4f99
update
eugene123tw Nov 27, 2024
8da83fa
update
eugene123tw Nov 27, 2024
2bd71ca
Update
eugene123tw Nov 28, 2024
a1da0e1
update recipes
eugene123tw Nov 28, 2024
1f2a9b3
Add d-fine-m
eugene123tw Nov 29, 2024
f5aa07c
Fix recipes
eugene123tw Nov 29, 2024
8bf6df2
Merge branch 'develop' into eugene/d-fine-poc
eugene123tw Dec 2, 2024
84baf23
dfine-l
eugene123tw Dec 2, 2024
adf6947
Add dfine m - no aug
eugene123tw Dec 3, 2024
3d92e7e
format changes
eugene123tw Dec 3, 2024
b91bb42
learnable params + disable teacher distillation
eugene123tw Dec 3, 2024
7daecda
update
eugene123tw Dec 3, 2024
8724219
add recipes
eugene123tw Dec 3, 2024
5e0f0f7
update
eugene123tw Dec 3, 2024
2140f68
update
eugene123tw Dec 3, 2024
71ce5d3
update recipes
eugene123tw Dec 4, 2024
175fc07
add dfine_hgnetv2_x
eugene123tw Dec 4, 2024
5b7f4aa
Update recipes
eugene123tw Dec 5, 2024
4d523f6
add tile DFine recipes
eugene123tw Dec 5, 2024
c40190a
update recipes and tile batch size
eugene123tw Dec 5, 2024
6b12cb3
update
eugene123tw Dec 5, 2024
090b2ca
Merge branch 'develop' into eugene/d-fine-poc
eugene123tw Dec 5, 2024
4381600
update LR
eugene123tw Dec 5, 2024
a305240
DFine revert LR changes
eugene123tw Dec 6, 2024
2149406
make multi-scale optional
eugene123tw Dec 6, 2024
62d63a0
update tile recipes
eugene123tw Dec 6, 2024
11679a9
update tiling recipes
eugene123tw Dec 9, 2024
f701446
add backbone pretrained weights
eugene123tw Dec 10, 2024
8dba5bd
updawte
eugene123tw Dec 10, 2024
7aa450f
update
eugene123tw Dec 11, 2024
4d340e2
loss
eugene123tw Dec 11, 2024
6f5fea7
update
eugene123tw Dec 12, 2024
ab81bc0
Update
eugene123tw Dec 15, 2024
04a10d0
refactor d-fine criterion
eugene123tw Dec 16, 2024
f9d7137
* Fix docstring punctuation and remove unused aux_loss parameter in D…
eugene123tw Dec 17, 2024
021cf2f
Update style changes
eugene123tw Dec 18, 2024
7ee9fa2
conv batchnorm fuse
eugene123tw Dec 18, 2024
a88a9d1
update hybrid encoder
eugene123tw Dec 19, 2024
edf22a0
Refactor DFINE HybridEncoderModule to improve code clarity and remove…
eugene123tw Dec 19, 2024
53c635d
minor update
eugene123tw Dec 19, 2024
fe1e7f9
Refactor D-FINE module structure by removing obsolete detector file a…
eugene123tw Dec 19, 2024
45d0360
Refactor import paths in D-FINE module and clean up unused code
eugene123tw Dec 19, 2024
7b4c899
Refactor D-FINE module by removing commented code, cleaning up import…
eugene123tw Dec 19, 2024
0d0f99b
Refactor D-FINE module by updating type hints, improving error messag…
eugene123tw Dec 20, 2024
64ea42a
Refactor D-FINE module by improving the weighting function's return s…
eugene123tw Dec 20, 2024
dff6ecc
Update d-fine unit test
eugene123tw Dec 20, 2024
b676b51
Refactor D-FINE module by enhancing docstrings for clarity and updati…
eugene123tw Dec 20, 2024
76089f1
Add D-Fine Detection Algorithm entries to CHANGELOG and object detect…
eugene123tw Dec 20, 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
2 changes: 2 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,8 @@ All notable changes to this project will be documented in this file.
(<https://github.com/openvinotoolkit/training_extensions/pull/4066>)
- Add GPU memory monitor hook
(<https://github.com/openvinotoolkit/training_extensions/pull/4118>)
- Add D-Fine Detection Algorithm
(<https://github.com/openvinotoolkit/training_extensions/pull/4142>)

### Enhancements

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,16 @@ We support the following ready-to-use model recipes:
+------------------------------------------------------------------------------------------------------------------------------------------------------------+---------------------+---------------------+-----------------+
| `Object_Detection_ResNeXt101_ATSS <https://github.com/openvinotoolkit/training_extensions/blob/develop/src/otx/recipe/detection/atss_resnext101.yaml>`_ | ResNeXt101-ATSS | 434.75 | 344.0 |
+------------------------------------------------------------------------------------------------------------------------------------------------------------+---------------------+---------------------+-----------------+
| `D-Fine N Detection <https://github.com/openvinotoolkit/training_extensions/blob/develop/src/otx/recipe/detection/dfine_n.yaml>` | D-Fine N | 7.165 | 14.8 |
+------------------------------------------------------------------------------------------------------------------------------------------------------------+---------------------+---------------------+-----------------+
| `D-Fine S Detection <https://github.com/openvinotoolkit/training_extensions/blob/develop/src/otx/recipe/detection/dfine_s.yaml>` | D-Fine S | 25.1714 | 39.9 |
+------------------------------------------------------------------------------------------------------------------------------------------------------------+---------------------+---------------------+-----------------+
| `D-Fine M Detection <https://github.com/openvinotoolkit/training_extensions/blob/develop/src/otx/recipe/detection/dfine_m.yaml>` | D-Fine M | 56.6844 | 75.4 |
+------------------------------------------------------------------------------------------------------------------------------------------------------------+---------------------+---------------------+-----------------+
| `D-Fine L Detection <https://github.com/openvinotoolkit/training_extensions/blob/develop/src/otx/recipe/detection/dfine_l.yaml>` | D-Fine L | 91.0344 | 120.0 |
+------------------------------------------------------------------------------------------------------------------------------------------------------------+---------------------+---------------------+-----------------+
| `D-Fine X Detection <https://github.com/openvinotoolkit/training_extensions/blob/develop/src/otx/recipe/detection/dfine_x.yaml>` | D-Fine X | 202.486 | 240.0 |
+------------------------------------------------------------------------------------------------------------------------------------------------------------+---------------------+---------------------+-----------------+

Above table can be found using the following command

Expand Down
Loading
Loading