YOLOv3_PyTorch

YOLOv3_PyTorch copied to clipboard

hello bobliu, I use the same code to train in coco2014, but I only get 34 ap using coco API can you share some detail about your training details? thank...

zhenghao977

darknet21 pretrained

2

I want to know Where can I download it

liuhengli

About the loss calculation

2

Maybe I think you could set self.mse_loss = nn.MSELoss() self.bce_loss = nn.BCELoss() to self.mse_loss = nn.MSELoss(size_average=False) self.bce_loss = nn.BCELoss(size_average=False) and then calculate losses like this way: n_mask = torch.sum(mask) n_noobj_mask...

LordLiang

大佬，我迁移了你的工程，loss也有下降，但是下降比较慢，而且训完比较多iter之后测试并没有结果，您能给我点意见吗？训的coco2017，不是14

linhezheng19

linhezheng19

zqxiang

Box overlapping & false positive issue

8

Hi, thanks for providing the code and script. :-) I trained with coco dataset using your training code. I only changed "parallels : [0,1,2,3]" to "paralles : [0]" in params.py...

doobidoob

Problem in grid generating

1

Hi, it occurs to me when i test on my own dataset sized at (1248, 416). ``` File "/home/kuro/dev/virtualenvs/pytorch37/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in __call__ result = self.forward(*input, **kwargs) File "/home/kuro/dev/projects/YOLOv3_PyTorch/test/../nets/yolo_loss.py", line...

Kuro96

my dataset，why?

1

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 15 and 16 in dimension 2 at c:\a\w\1\s\tmp_conda_3.7_110206\conda\conda-bld\pytorch_1550401474361\work\aten\src\thc\generic/THCTensorMath.cu:83

skysunsun

Traceback (most recent call last): File "test_images.py", line 158, in main() File "test_images.py", line 154, in main test(config) File "test_images.py", line 126, in test plt.text(x1, y1, s=classes[int(cls_pred)], color='white', IndexError: list...

skysunsun