Update modular_rtdetrv2.py
Browse files- modular_rtdetrv2.py +1 -0
modular_rtdetrv2.py
CHANGED
|
@@ -334,6 +334,7 @@ class RTDetrV2HungarianMatcher(RTDetrHungarianMatcher):
|
|
| 334 |
pass
|
| 335 |
|
| 336 |
|
|
|
|
| 337 |
class RTDetrV2ForObjectDetection(RTDetrForObjectDetection):
|
| 338 |
pass
|
| 339 |
|
|
|
|
| 334 |
pass
|
| 335 |
|
| 336 |
|
| 337 |
+
# must inherit the new classes!
|
| 338 |
class RTDetrV2ForObjectDetection(RTDetrForObjectDetection):
|
| 339 |
pass
|
| 340 |
|