Skip to content

更新对于 yolov4 的 indice_merge() 函数 #83

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

更新对于 yolov4 的 indice_merge() 函数 #83

wants to merge 2 commits into from

Conversation

@chumingqian
Copy link

@chumingqian chumingqian commented Nov 4, 2020 •
edited
Loading

由于v3 和 V4 的 cfg 文件中， yolo 层出现的顺序正好相反。
@chumingqian
Update utils.py
085cc7b 
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) ) 的功能， 这样做的目的是什么。
@chumingqian
Update utils.py
fd8214a
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

1 participant

@chumingqian