Skip to content

Commit 76b5044

Browse files
authored
Enable keypoint detection training through config conversion (#4034)
enable keypoint det config converter
1 parent a3c9fc6 commit 76b5044

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

src/otx/tools/converter.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,15 @@
183183
"task": OTXTaskType.ANOMALY_SEGMENTATION,
184184
"model_name": "stfpm",
185185
},
186+
# KEYPOINT_DETECTION
187+
"Custom_Keypoint_Detection_Rtmpose_T": {
188+
"task": OTXTaskType.KEYPOINT_DETECTION,
189+
"model_name": "rtmpose_tiny",
190+
},
191+
"Custom_Keypoint_Detection_Rtmpose_T_Single_Obj": {
192+
"task": OTXTaskType.KEYPOINT_DETECTION,
193+
"model_name": "rtmpose_tiny_single_obj",
194+
},
186195
}
187196

188197

0 commit comments

Comments
 (0)