convert to onnx error - Githubissues

chenjun2hao / CenterFace.pytorch

unofficial version of centerface, which achieves the best balance between speed and accuracy at face detection

236 stars 60 forks source link

convert to onnx error #11

Closed ForestWang closed 4 years ago

ForestWang commented 4 years ago

Hi Sir: when converting to onnx, got the error: "RuntimeError: Only tuples, lists and Variables supported as JIT inputs, but got dict" could you figure it out?

many thanks.

chenjun2hao commented 4 years ago

you should uncommit this line # @dict2list # 转onnx的时候需要将输出由dict转成list模式 in centerface_mobilenet_v2_fpn.py