yolo2-pytorch
yolo2-pytorch copied to clipboard
YOLOv2 in PyTorch
``` iou_penalty = 0 - iou_pred_np[best_ious < cfg.iou_thresh] _iou_mask[best_ious
1. When running train.py,how can I select gpu_id,for 0,1,2....? 2. What does batch_size and train_batch_size mean? Thank you for your help!!!
Thanks in advance.
i train my datasets , but it can only run two steps and occurred----- ValueError: zero-size array to reduction operation maximum which has no identity. what can i do ?...
emm, I have question about train cls_loss ,when I train on my own data ,the class loss can't down in 0.999,I check my data, and even change the learnning rate...
Today, I first get this code. I want to train the yolov2 at the VOC2007 dataset. Then, I get this errror. $ python train.py voc_2007_trainval gt roidb loaded from /home/lab603/pythonproject/yolo2-pytorch-master/data/cache/voc_2007_trainval_gt_roidb.pkl...
https://github.com/YunYang1994/tensorflow-yolov3 hope it helps you
I've trained yolov3 on mobilenet backend and get a good MAP(about 90M and 2 times faster than keras yolov3)。I upload my code for mobilenet-yolov3 on https://github.com/Adamdad/keras-YOLOv3-mobilenet.git use it as you...
mAP
Have you ever evaluate the transformed trained model in VOC2007? I've tried your code and got a 71.9 mAP while the original is 76.8. Then I found a tiny error...