yolo2-pytorch
yolo2-pytorch copied to clipboard
longcw
YOLOv2 in PyTorch
Hi, When I try the demo I always get `Traceback (most recent call last): File "demo.py", line 63, in bbox_pred, iou_pred, prob_pred, image.shape, cfg, thresh) File "/export/mlrg/aelnouby/projects/3rdParty/yolo2-pytorch/utils/yolo.py", line 108, in...
@longcw @cory8249 @crazylyf I am planing to train the YOLO2 on the SynthText dataset, so when I prepare the dataset, I need to resize the image into 416*416 and rearrange...
CUDA_PATH set in make.sh but CUDAHOME used in utils/build.py
python 3.6 torch 0.3.0 When I run the train.py lxt@slave-01:~/pytorch/yolo2-pytorch$ python train.py Traceback (most recent call last): File "train.py", line 8, in from darknet import Darknet19 File "/home/lxt/pytorch/yolo2-pytorch/darknet.py", line 9,...
Hello, I am trying to build the project by executing ./make.sh But I'm having the following error locating the file roi_pooling_kernel.cu.o I am using: gcc 4.9 CUDA Version 8.0.61 CudNN...
The instructions for using custom datasets say: "The four values in each row should correspond to x_bottom_left, y_bottom_left, x_top_right, and y_top_right" However, the tags in the VOC dataset appear to...
Good work. But I am confused about how to calculate cls loss. It seems that you used MSELoss in [your code](https://github.com/longcw/yolo2-pytorch/blob/master/darknet.py#L222). However, I find that in darknet, when computing gradient,...
I would like to retrain the network on a custom dataset of food items. I'm sort of new to this. Are there instructions about how to do it? I am...
Should we be normalizing the inputs prior to feeding them into the network [[1](https://stats.stackexchange.com/questions/211436/why-do-we-normalize-images-by-subtracting-the-datasets-image-mean-and-not-the-c)]? All of the PyTorch networks contained in the model zoo use the same mean and std...
Hello I am trying to train YOLO using VOC dataset. I follow your instruction. I have successfully run and save the model, but when I entering epoch 5 in iteration...
