4 ms·
There is a pytorch port from Ultralytics (https://github.com/ultralytics/yolov3 https://github.com/ultralytics/yolov3). Nobody seems to have figured out how to
by joshvm 8y ago
There is a pytorch port from Ultralytics (https://github.com/ultralytics/yolov3 https://github.com/ultralytics/yolov3). Nobody seems to have figured out how to achieve the training performance of darknet though, which is entirely uncommented C. The source is all there, but the loss function changed between v2 and v3, and its not documented in the paper. I think it's been fixed in that pytorch port now though. The only frustrating thing is that every commit in the repo is called update...
Alternatively... you can train in darknet and then run inference in another framework of choice.
Also shameless plug: I wrote an annotation tool which is designed to output darknet formatted labels: https://github.com/jveitchmichaelis/deeplabel https://github.com/jveitchmichaelis/deeplabel
- syntaxing 8y agoYeah, I don't remember where I read it but it took them a couple weeks to train the model from scratch. I tried training my own weights by scratch it was practically impossible using a Tesla K80. But it's pleasantly surprising how good the transfer learning results are on a custom data set. You can get some "state of the art" results when you train for a couple hours. It's really impressive how he came up with YOLO and wrote his own deep learning library from scratch. Thank you for the links! I'm going to check both out. I want to see if the PyTorch port works with the new deployment feature from 1.0.