DengpanFu / LUPerson

Unsupervised Pre-training for Person Re-identification (LUPerson)
242 stars 35 forks source link

Errors in converting the torch models to ONNX #5

Open JosephAssaker opened 3 years ago

JosephAssaker commented 3 years ago

Following the steps mentioned in LUPerson/fast-reid/tools/deploy/ to export an ONNX file, and after running the below command (which I'm not 100% sure I got right):

!python onnx_export.py --config-file ../../configs/CMDM/mgn_R50_moco.yml --name "baseline_R50" --output outputs/onnx_model --opts MODEL.WEIGHTS market_mgn_r50.pth

I am getting the following error: AttributeError: module 'onnx' has no attribute 'optimizer'

Any help would be appreciated, and possibly a clarification as to which exact yml and pth files should be used in order to export the MGN model trained from LUPerson pre-trained networks.