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.
You can use pretrained versions of these models.
Here is a very good 3-loc example straight from pytorch hub:
Thanks a lot, appreciate it!!