Some more details on modified yolo about redtail HOT 4 OPEN

Hi Marc,
The way we trained our version of the model was very similar to the original YOLO v1 experiments:

1. Train slightly modified version of YOLO in darknet. The reason we had to change the topology was TensorRT did not support some of the features (like Leaky ReLUs or untied convolutions) at that time. I've added the darknet config file here.
2. Convert the resulting weights file to Caffe format. The script that we wrote was somewhat hacky :) so we decided not publish it until we make it pretty. However, there are darknet -> Caffe converters like this which hopefully should work.

from redtail.

Hi Alexey,

Thanks a lot! This was very helpful and I think this should work for what we want to test.

If you have any performance metrics to compare TensorRT vs the darknet implementation that would also be very interesting to see. We will run our own benchmark tests anyways, though, so no problem if you can't release them.

Thanks!
marc

from redtail.

Hi Alexey,

Would you have the original darknet weights file available for reference?

Thanks!
Marc

from redtail.

Hi Marc,
I've uploaded the weights to my branch (alexeyk/yolo), please do git pull. We will merge it into master in the next release along with more YOLO-related info.

from redtail.