Skip to content

v0.4.0

Compare
Choose a tag to compare
@RangiLyu RangiLyu released this 08 Jun 03:03
· 121 commits to main since this release

What's new in v0.4.0

  1. Fix a little bug in demo.py by BlainWu (#210)
  2. Add script to export TorchScript model by strawberrypie (#211)
  3. Use fixed output names when exporting ONNX (#218)
  4. Use scale_factor instead of fixed size in resize to support dynamic shape inference (#218)
  5. Ensure num_classes equal len(class_names) by ZHEQIUSHUI (#221)
  6. Fix a bug in mnn demo while using GPU device by AcherStyx (#234)
  7. Fix with_last_conv bug in shufflenet (#239)
  8. Support batch eval (#241)
  9. Add nanodet-m-1.5x models (#242)
  10. Update model benchmark (#246)
  11. Prevent lightning Trainer from disabling cudnn.benchmark (#249)
  12. Fix multi-GPU evaluation bug with pytorch-lightning (#254)

Download pretrained models

Model Backbone Resolution COCO mAP FLOPS Params Pre-train weight
NanoDet-m ShuffleNetV2 1.0x 320*320 20.6 0.72B 0.95M Download
NanoDet-m-416 ShuffleNetV2 1.0x 416*416 23.5 1.2B 0.95M Download
NanoDet-m-1.5x ShuffleNetV2 1.5x 320*320 23.5 1.44B 2.08M Download
NanoDet-m-1.5x-416 ShuffleNetV2 1.5x 416*416 26.8 2.42B 2.08M Download
NanoDet-t ShuffleNetV2 1.0x 320*320 21.7 0.96B 1.36M Download
NanoDet-g Custom CSP Net 416*416 22.9 4.2B 3.81M Download
NanoDet-EfficientLite EfficientNet-Lite0 320*320 24.7 1.72B 3.11M Download
NanoDet-EfficientLite EfficientNet-Lite1 416*416 30.3 4.06B 4.01M Download
NanoDet-EfficientLite EfficientNet-Lite2 512*512 32.6 7.12B 4.71M Download
NanoDet-RepVGG RepVGG-A0 416*416 27.8 11.3B 6.75M Download

Download ncnn models below