timmh / MegaDetectorLite

MegaDetector 5.0 for Embedded Devices

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MegaDetectorLite

This repository provides scripts to convert the weights of MegaDetector 5.0 into the ONNX and TensorRT formats to simplify the usage in embedded devices. yolort further optimizes the runtime performance by embedding the pre- and post-processing steps into the models.

Converted Weights

Already converted weights can be found under releases.

TODOs

  • enable batch sizes larger than 1 and flexible spatial dimensions for the TensorRT models
  • perform benchmarks on some embedded devices
  • automatically compare detection performance of derived models

Acknowledgments

The test.jpg file is provided by the Caltech Camera Traps dataset under the terms of the Community Data License Agreement (permissive variant).

About

MegaDetector 5.0 for Embedded Devices

License:MIT License


Languages

Language:Python 100.0%