mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 07:06:36 +00:00
[AZ-171] Remove unused onnx_batch config, add macOS skip for CoreML tests
Made-with: Cursor
This commit is contained in:
@@ -38,7 +38,6 @@ class TrainingConfig(BaseModel):
|
||||
|
||||
class ExportConfig(BaseModel):
|
||||
onnx_imgsz: int = 1280
|
||||
onnx_batch: int = 4
|
||||
|
||||
|
||||
class Config(BaseModel):
|
||||
|
||||
+1
-4
@@ -25,9 +25,7 @@ def export_rknn(model_path):
|
||||
pass
|
||||
|
||||
|
||||
def export_onnx(model_path, batch_size=None):
|
||||
if batch_size is None:
|
||||
batch_size = constants.config.export.onnx_batch
|
||||
def export_onnx(model_path):
|
||||
model = YOLO(model_path)
|
||||
onnx_path = Path(model_path).stem + '.onnx'
|
||||
if path.exists(onnx_path):
|
||||
@@ -36,7 +34,6 @@ def export_onnx(model_path, batch_size=None):
|
||||
model.export(
|
||||
format="onnx",
|
||||
imgsz=constants.config.export.onnx_imgsz,
|
||||
batch=batch_size,
|
||||
dynamic=True,
|
||||
simplify=True,
|
||||
nms=True,
|
||||
|
||||
Reference in New Issue
Block a user