search

ShiqiYu / libfacedetection.train

The training program for libfacedetection for face detection and 5-landmark detection.
Apache License 2.0
766 stars 209 forks source link

optimize onnx export #68

Closed Wwupup closed 1 year ago

Wwupup commented 1 year ago
  1. Replace shape with scale_factor' in F.interpolate() of tfpn.py.
  2. Decouple the batch_size when exporting onnx in yunet_head.py.
  3. Fix bug of dynamic_axes in yunet2onnx.py;