[models] Add keypoint detection models

pyronear / pyro-vision

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

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

Apache License 2.0

51 stars 24 forks source link

[models] Add keypoint detection models #124

Open frgfm opened 3 years ago

frgfm commented 3 years ago

Currently we perform keypoint detection by:

using a robust image classifier
use its class activation to extrapolate the keypoint

We should explore alternative ways with light backbones to perform actual keypoint detection and train it properly.

Aym98 commented 8 months ago

Is this issue still relevant ?

frgfm commented 7 months ago

Hey @Aym98 :wave:

Yes, so far we've defaulted to object detection, but this is still not ideal. Add comparatively, keypoint detection is a much more adequate task to model. We could train much lighter models for this :ok_hand: