@@ -58,9 +58,6 @@ def _get_processor_function(model_type: str) -> Callable:
58
58
if "yolonas" in model_type :
59
59
return _process_yolonas
60
60
61
- if "yolov12" in model_type :
62
- return _process_yolov12
63
-
64
61
if "rfdetr" in model_type :
65
62
return _process_rfdetr
66
63
@@ -227,46 +224,6 @@ def _process_yolo(model_type: str, model_path: str, filename: str) -> str:
227
224
return zip_file_name
228
225
229
226
230
- def _process_yolov12 (model_type : str , model_path : str , filename : str ) -> str :
231
- # For YOLOv12, since it uses a special Ultralytics version,
232
- # state dict extraction and model artifacts are handled during model conversion
233
-
234
- print (
235
- "Note: Model must be trained using ultralytics from https://github.com/sunsmarterjie/yolov12 "
236
- "or through the Roboflow platform"
237
- )
238
-
239
- # Check if model_path exists
240
- if not os .path .exists (model_path ):
241
- raise FileNotFoundError (f"Model path { model_path } does not exist." )
242
-
243
- # Find any .pt file in model path
244
- model_files = os .listdir (model_path )
245
- pt_file = next ((f for f in model_files if f .endswith (".pt" )), None )
246
-
247
- if pt_file is None :
248
- raise RuntimeError ("No .pt model file found in the provided path" )
249
-
250
- # Copy the .pt file to weights.pt if not already named weights.pt
251
- if pt_file != "weights.pt" :
252
- shutil .copy (os .path .join (model_path , pt_file ), os .path .join (model_path , "weights.pt" ))
253
-
254
- required_files = ["weights.pt" ]
255
-
256
- optional_files = ["results.csv" , "results.png" , "model_artifacts.json" ]
257
-
258
- zip_file_name = "roboflow_deploy.zip"
259
- with zipfile .ZipFile (os .path .join (model_path , zip_file_name ), "w" ) as zipMe :
260
- for file in required_files :
261
- zipMe .write (os .path .join (model_path , file ), arcname = file , compress_type = zipfile .ZIP_DEFLATED )
262
-
263
- for file in optional_files :
264
- if os .path .exists (os .path .join (model_path , file )):
265
- zipMe .write (os .path .join (model_path , file ), arcname = file , compress_type = zipfile .ZIP_DEFLATED )
266
-
267
- return zip_file_name
268
-
269
-
270
227
def _process_rfdetr (model_type : str , model_path : str , filename : str ) -> str :
271
228
_supported_types = ["rfdetr-base" , "rfdetr-large" ]
272
229
if model_type not in _supported_types :
0 commit comments