longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

single class training issus #113

Open riyaj8888 opened 5 years ago

riyaj8888 commented 5 years ago

when i am training yolov2 for single class i am getting following error. " net_utils.load_net(args.model, net) File "/home/ubuntu/projects/lesiondetectlong/utils/network.py", line 63, in loadnet v.copy(param) RuntimeError: The expanded size of the tensor (30) must match the existing size (60) at non-singleton dimension 0 " when i printed out the output channels of conv5 layers it has to show "out_channels = num_anchors * (num_classes + 5) " for single class it must be 30 . but i am getting 60 output channels