acm5656 / ssd_pytorch Goto Github PK

为什么大哥的sk从15开始 不是应该30开始吗

Traceback (most recent call last):
File "F:/use_pytorch_SSD_data/ssd_pytorch-master_simple/ssd_pytorch-master/Test.py", line 59, in
while detections[0,i,j,0] >= 0.5:
IndexError: index 200 is out of bounds for dimension 2 with size 200
分类改为两类，修改了detect = Detect(config.class_num, 0, 200, 0.5, 0.65)和 while detections[0,i,j,0] >= 0.5:会报错，求怎么解决？

when the iteration=120170,this error was occured and the loss is not too small .
as follows:
epoch : 231 iter : 119990 step : 332 loss : 2.7521151838631464
epoch : 231 iter : 120000 step : 342 loss : 2.3202000970709813
epoch : 231 iter : 120010 step : 352 loss : 2.2625636844844608
epoch : 231 iter : 120020 step : 362 loss : 2.1100341796875
epoch : 231 iter : 120030 step : 372 loss : 2.1073843819754465
epoch : 231 iter : 120040 step : 382 loss : 2.1795616149902344
epoch : 231 iter : 120050 step : 392 loss : 2.4456642951754577
epoch : 231 iter : 120060 step : 402 loss : 2.0380105777662627
epoch : 231 iter : 120070 step : 412 loss : 2.336949090537785
epoch : 231 iter : 120080 step : 422 loss : 2.598242623465402
epoch : 231 iter : 120090 step : 432 loss : 2.424121765863328
epoch : 231 iter : 120100 step : 442 loss : 2.4087103054608185
epoch : 231 iter : 120110 step : 452 loss : 2.2298266662293997
epoch : 231 iter : 120120 step : 462 loss : 1.706140349893009
epoch : 231 iter : 120130 step : 472 loss : 2.384180556248574
epoch : 231 iter : 120140 step : 482 loss : 2.154784760944584
epoch : 231 iter : 120150 step : 492 loss : 2.1820303144909086
epoch : 231 iter : 120160 step : 502 loss : 2.0698028564453126
epoch : 231 iter : 120170 step : 512 loss : 2.167848598457382
(M2Det) xlm@xlm-OptiPlex-9010:/ssd_pytorch-master$ 000.
000.: command not found
(M2Det) xlm@xlm-OptiPlex-9010:/ssd_pytorch-master$
(M2Det) xlm@xlm-OptiPlex-9010:/ssd_pytorch-master$
(M2Det) xlm@xlm-OptiPlex-9010:/ssd_pytorch-master$

这里为何要用unsqueeze呀