pytorch-yolo-v3 icon indicating copy to clipboard operation
pytorch-yolo-v3 copied to clipboard

simplify create_modules() 'route' option

Open Anner-deJong opened this issue 6 years ago • 0 comments

Thanks a million for your blogs! Im currently going through the entire setup and loving it :D I thought I'd give back a bit by (starting with) this simplification. Correct me if I'm wrong, but I think all you are trying to achieve is to sum the number of output filters for the 1 or 2 layers a route-layer is referring to. This can be done in 1 line as proposed.

I'm not a 100% sure if Redmon is using 0 indexing when referring to his layers, but I believe he is.

Anner-deJong avatar Oct 01 '18 09:10 Anner-deJong