pyronear / pyro-vision

Computer vision library for wildfire detection 🌲 Deep learning models in PyTorch & ONNX for inference on edge devices (e.g. Raspberry Pi)

Home Page:https://pyronear.org/pyro-vision/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[models] Add an object detection model

frgfm opened this issue · comments

The latest torchvision releases include highly optimized models for object detection and semantic segmentation. Using our keypoint annotations, we could extrapolate a small bounding box around the keypoint and train a FasterRCNN with a MobileNetV3 backbone.