SpursLipu / YOLOv3v4-ModelCompression-MultidatasetTraining-Multibackbone

YOLO ModelCompression MultidatasetTraining
GNU General Public License v3.0
444 stars 136 forks source link

Update utils.py #81

Closed chumingqian closed 3 years ago

chumingqian commented 3 years ago

下午好, 李: 1:b_temp = (torch.ones(int(2 * (5 - j) / scale - 1)) b[i]).long().cuda() 增加了. cuda() 参数, 避免了 find .cpu , .gpu 报错; 2: 不使用了 indice_merge() 函数, 因为原始了 indice[0] 中 有四个变量 , indices[0]: [ [0,0,3] , [1,2,2], [2,2,6] [16,16,24] ] 经过indices_merge() 函数, (2( 5- j) 之后, 出现[16,16, 832] 会出现 index 832 out of 608 .的错误; 3: 暂时 没有读懂, indices_merge() 函数 为什么 将temp[2] , temp[3] 乘上 (2 (5-j) ) 的功能, 这样做的目的是什么。