Commit 0c04f88
authored
Add D-FINE to KerasHub (#2318)
* init: Add initial project structure and files
* test: Enable test cases (will fail until HGNetV2 dep added)
* refactor: Make clean and consistent API design choices
* refactor: Remove the task model from the scope of this PR
* refactor: Enhance test suite robustness and standardize weight init to match original
* nit: Remove the problematic channel_axis from the deserialization args
* nit: Replace hyphen with underscore in preset name
* refactor: Implement code cleanup based on review feedback
* refactor: Address code reviews
* nit: Remove unnecessary Keras serialization decorator
* refactor: Resolve review comments
* update: Enable HGNetV2 and D-FINE tests!
* test: Enable `run_data_format_check`
* feat: Add DFineObjectDetector (TODO: Make loss batch-aware)
* skipif: Bbox utils are not supported before Keras < 3.8.0
* test: Minor implementation review
* Complete TODO: Make loss fully batch-aware
* finals: Add adaptation explanations; denoising query range
* fix: GPU tests + memory optimization
* change: Reduce memory usage with minimal test configuration
* reviews: Incorporate comments
* nits: Move loss to separate file + address review feedback (Gemini, Sachin)
* nit: Skip HF preset loading test on TensorFlow GPU CI due to an unknown OOM error1 parent d625954 commit 0c04f88
File tree
25 files changed
+8979
-17
lines changed- keras_hub
- api
- layers
- models
- src
- models
- d_fine
- hgnetv2
- tests
- utils/transformers
- tools/checkpoint_conversion
25 files changed
+8979
-17
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
75 | 75 | | |
76 | 76 | | |
77 | 77 | | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
78 | 81 | | |
79 | 82 | | |
80 | 83 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
108 | 108 | | |
109 | 109 | | |
110 | 110 | | |
| 111 | + | |
| 112 | + | |
| 113 | + | |
| 114 | + | |
| 115 | + | |
| 116 | + | |
| 117 | + | |
| 118 | + | |
| 119 | + | |
111 | 120 | | |
112 | 121 | | |
113 | 122 | | |
| |||
Whitespace-only changes.
0 commit comments