microsoft / arcticseals

A deep learning project in cooperation with the NOAA Marine Mammal Lab to detect & classify arctic seals in aerial imagery to understand how they’re adapting to a changing world.
MIT License
33 stars 14 forks source link

Object detection approaches #6

Open Marcel-Simon opened 6 years ago

Marcel-Simon commented 6 years ago

Another way to contribute is training an actual object detection model, which does not require the hotspot detections. The input is an image, the output is a set of bounding box. I would consider this more difficult then the hotspot classification, but I would be happy to work with people on this task as well. First ideas for action items:

Athapan commented 6 years ago

I got the tensorflow object detection working at one point. Someone mentioned that Azure release an objection detection that work quite well when they tried it.

https://docs.microsoft.com/en-us/azure/cognitive-services/custom-vision-service/python-tutorial-od

I'll see if I can sometime to experiment with these.

Marcel-Simon commented 6 years ago

I added a detection code based on PyTorch so people can get started quickly: https://github.com/Microsoft/arcticseals/tree/master/src/SealDetectionRCNN please refer to the README.md on how to get everything running. We will soon also upload a nice dataset of all the labels we have.