Learning Transfer or Fine tuning Yolo or MobilenetV2

vision
1

Hi
I was wondering if it is possible to train YOLOv2 or MobileNetV2 with PyTorch?
I’m looking into building a real-time webcam object detector and do transfer learning to teach it to recognize specific objects of my own.
These two would be perfect for my application.
Thanks!

2

You can use pretrained versions of these models.

Here is a very good 3-loc example straight from pytorch hub:

3

Thanks a lot, appreciate it!!