자체 개발한 OCR검출기 모델 onnx모델 변환 및 양자화를 진행하고 싶은데 warboy-vision make-model --config_file “./ocr_det.yaml” 명령어를 실행시키니 아래와 같은 에러가 발생합니다.
(warboy) clapi@sysadmin-B7105F48TV4HR-2T-N:~/warboy-vision-models/src/warboy/cfg/model_config/object_detection$ warboy-vision make-model --config_file “./ocr_det.yaml”
Warning: The model is not a YOLO model. The need_edit option is ignored.
Load PyTorch Model from ./models/freeze_detector.pt…
Traceback (most recent call last):
File “/home/clapi/miniconda3/envs/warboy/bin/warboy-vision”, line 8, in
sys.exit(cli())
File “/home/clapi/.local/lib/python3.10/site-packages/click/core.py”, line 1442, in call
return self.main(*args, **kwargs)
File “/home/clapi/.local/lib/python3.10/site-packages/click/core.py”, line 1363, in main
rv = self.invoke(ctx)
File “/home/clapi/.local/lib/python3.10/site-packages/click/core.py”, line 1830, in invoke
return _process_result(sub_ctx.command.invoke(sub_ctx))
File “/home/clapi/.local/lib/python3.10/site-packages/click/core.py”, line 1226, in invoke
return ctx.invoke(self.callback, **ctx.params)
File “/home/clapi/.local/lib/python3.10/site-packages/click/core.py”, line 794, in invoke
return callback(*args, **kwargs)
File “/home/clapi/miniconda3/envs/warboy/lib/python3.10/site-packages/src/cli/make_model/run_make_model.py”, line 35, in run_make_model
onnx_tools.export_onnx(need_edit=need_edit)
File “/home/clapi/miniconda3/envs/warboy/lib/python3.10/site-packages/src/warboy/tools/onnx_tools.py”, line 39, in export_onnx
raise NotImplementedError(
NotImplementedError: Export ONNX for object_detection >> freeze_detector is not implemented yet!