convert_onnx error
vanillar7 opened this issue · comments
[08:22:52] INFO - Preparing 'arcface_r100_v1' model...
[08:22:52] INFO - Building TRT engine for arcface_r100_v1...
mxnet version: 1.7.0
onnx version: 1.7.0
Traceback (most recent call last):
File "prepare_models.py", line 45, in
prepare_models()
File "prepare_models.py", line 40, in prepare_models
prepare_backend(model_name=model, backend_name=backend_name, im_size=max_size, force_fp16=force_fp16,
File "/app/modules/model_zoo/getter.py", line 126, in prepare_backend
convert_onnx(temp_onnx_model,
TypeError: 'NoneType' object is not callable
Starting InsightFace-REST using 1 workers.
[08:22:54] INFO - Reshaping ONNX inputs to: (1, 3, 640, 640)
mxnet version: 1.7.0
onnx version: 1.7.0
Traceback (most recent call last):
File "/usr/local/bin/uvicorn", line 8, in
sys.exit(main())
File "/usr/local/lib/python3.8/dist-packages/click/core.py", line 829, in call
return self.main(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/click/core.py", line 782, in main
rv = self.invoke(ctx)
File "/usr/local/lib/python3.8/dist-packages/click/core.py", line 1066, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/usr/local/lib/python3.8/dist-packages/click/core.py", line 610, in invoke
return callback(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/uvicorn/main.py", line 362, in main
run(**kwargs)
File "/usr/local/lib/python3.8/dist-packages/uvicorn/main.py", line 386, in run
server.run()
File "/usr/local/lib/python3.8/dist-packages/uvicorn/server.py", line 48, in run
loop.run_until_complete(self.serve(sockets=sockets))
File "/usr/lib/python3.8/asyncio/base_events.py", line 616, in run_until_complete
return future.result()
File "/usr/local/lib/python3.8/dist-packages/uvicorn/server.py", line 55, in serve
config.load()
File "/usr/local/lib/python3.8/dist-packages/uvicorn/config.py", line 308, in load
self.loaded_app = import_from_string(self.app)
File "/usr/local/lib/python3.8/dist-packages/uvicorn/importer.py", line 20, in import_from_string
module = importlib.import_module(module_str)
File "/usr/lib/python3.8/importlib/init.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "", line 1014, in _gcd_import
File "", line 991, in _find_and_load
File "", line 975, in _find_and_load_unlocked
File "", line 671, in _load_unlocked
File "", line 783, in exec_module
File "", line 219, in _call_with_frames_removed
File "./app.py", line 34, in
processing = Processing(det_name=configs.models.det_name, rec_name=configs.models.rec_name,
File "./modules/processing.py", line 121, in init
self.model = FaceAnalysis(det_name=det_name, rec_name=rec_name, ga_name=ga_name, device=device,
File "./modules/face_model.py", line 69, in init
self.det_model = Detector(det_name=det_name, device=device, max_size=self.max_size,
File "./modules/face_model.py", line 35, in init
self.retina = get_model(det_name, backend_name=backend_name, force_fp16=force_fp16, im_size=max_size,
File "./modules/model_zoo/getter.py", line 175, in get_model
model_path = prepare_backend(model_name, backend_name, im_size=im_size, max_batch_size=max_batch_size, config=config, force_fp16=force_fp16,
File "./modules/model_zoo/getter.py", line 115, in prepare_backend
model = onnx.load(onnx_path)
File "/usr/local/lib/python3.8/dist-packages/onnx/init.py", line 114, in load_model
s = _load_bytes(f)
File "/usr/local/lib/python3.8/dist-packages/onnx/init.py", line 30, in _load_bytes
with open(cast(Text, f), 'rb') as readable:
FileNotFoundError: [Errno 2] No such file or directory: '/models/onnx/retinaface_r50_v1/retinaface_r50_v1.onnx'
@vanillar7 : Are you able to find retinaface_r50_v1.onnx file inside your docker container at path
**
/models/onnx/retinaface_r50_v1/retinaface_r50_v1.onnx
**
I have tested building docker image without cache, and indeed, now this bug occurs.
It's caused by update of upstream onnx-tensorrt library.
I'll push update once I get back to PC.
For now you can just swap lines 14 and 16 in dockerfile, and it'll solve the problem.