levihsu / OOTDiffusion

Official implementation of OOTDiffusion: Outfitting Fusion based Latent Diffusion for Controllable Virtual Try-on
Other
5.82k stars 838 forks source link

parsing_atr.onnx failed:Protobuf parsing failed. #118

Closed keloniton closed 8 months ago

keloniton commented 8 months ago

Hi. I get the following error. Any one to help? I will really appreciate it . Thanks @levihsu @ShineChen1024 @T-Gu Traceback (most recent call last): File "/home/jovyan/workspace/OOTDiffusion/run/run_ootd.py", line 30, in parsing_model = Parsing(args.gpu_id) File "/home/jovyan/workspace/OOTDiffusion/preprocess/humanparsing/run_parsing.py", line 20, in init self.session = ort.InferenceSession(os.path.join(Path(file).absolute().parents[2].absolute(), 'checkpoints/humanparsing/parsing_atr.onnx'), File "/opt/saturncloud/envs/saturn/lib/python3.10/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 419, in init self._create_inference_session(providers, provider_options, disabled_optimizers) File "/opt/saturncloud/envs/saturn/lib/python3.10/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 472, in _create_inference_session sess = C.InferenceSession(session_options, self._model_path, True, self._read_config_from_model) onnxruntime.capi.onnxruntime_pybind11_state.InvalidProtobuf: [ONNXRuntimeError] : 7 : INVALID_PROTOBUF : Load model from /home/jovyan/workspace/OOTDiffusion/checkpoints/humanparsing/parsing_atr.onnx failed:Protobuf parsing failed.

T-Gu commented 8 months ago

@keloniton Please check your downloaded onnx models, e.g. compare the file size with the original.

Rumois commented 8 months ago

@keloniton请检查您下载的 onnx 模型,例如将文件大小与原始文件大小进行比较。

Where to download onnx models

levihsu commented 8 months ago

@keloniton请检查您下载的 onnx 模型,例如将文件大小与原始文件大小进行比较。

Where to download onnx models

https://huggingface.co/levihsu/OOTDiffusion/tree/main/checkpoints/humanparsing