yolo2-pytorch
yolo2-pytorch copied to clipboard
longcw
YOLOv2 in PyTorch
Hi, When I read the code, I found something hard to understand: 1. In the code train.py, you randomly chose a scale index for training image every cfg.disp_interval. (https://github.com/longcw/yolo2-pytorch/blob/master/train.py#L120) 2....
Hi there, sorry about that to post this here, i was trying to write yolo in torch but struggled the backward function for so many days, but the gradients are...
Looking at the loss code https://github.com/longcw/yolo2-pytorch/blob/1b320fa/darknet.py#L243 I see that each loss term is divided by the number of groundtruth boxes in the batch. Is this correct? I don't see anything...
Ask for help,in your code,you use _ious[cell_ind, a, :] = ious_reshaped[cell_ind, a, i], I think _ious[cell_ind, a, :] = anchor_ious[a, i]
Dear @longcw, The [new version of YOLO](https://pjreddie.com/media/files/papers/YOLOv3.pdf) has been released (_i.e.,_ [YOLOv3](https://pjreddie.com/darknet/yolo/)). Can I use YOLOv3 cfg/weights with your code correctly?
Having this error after successfully executing the make shell file. File "demo.py", line 6, in from darknet import Darknet19 File "/home/suyash/AIML/yolo2-pytorch/darknet.py", line 8, in from layers.reorg.reorg_layer import ReorgLayer File "/home/suyash/AIML/yolo2-pytorch/layers/reorg/reorg_layer.py",...
Any suggestions for resolving this issue? I am using CUDA 9.0 on an AWS server
Hi , Generator_anchor_box file generate 10 values of anchors , i have question about these values , as we have 5 anchors and this generator generate 10 values, more likely...
In the code, the ground truth value for iou prediction is the iou between the anchor and the ground truth box (https://github.com/cory8249/yolo2-pytorch/blob/master/darknet.py#L111). But the paper says that " the objectness...
Hi Has anyone tried writing a data loader using torch.utils?? I'm wondering what do I need to change in order to do so.
