portugueslab / stytra

A modular package to control stimulation and track behaviour
http://www.portugueslab.com/stytra/
GNU General Public License v3.0
41 stars 26 forks source link

YoLo #46

Closed mariakesa closed 3 years ago

mariakesa commented 3 years ago

Hello:-),

Thank you so much for the wonderful library!

Does stryta use the YoLo algorithm from deep learning anywhere internally? If not would there be interest in developing a module with YoLo?

Maria

vilim commented 3 years ago

You're welcome! No, the tracking algorithms included in this library are not using deep learning (as it's not necessary and usually not fast enough for our tracking scenarios). If you want to add a new tracking algorithm that uses YOLO or another object detection network, have a look at examples for custom tracking methods here and here