-
Notifications
You must be signed in to change notification settings - Fork 28
Open
Description
这里的top取值不应该是 top = row_ind+j*length 。这会导致在i=1时,在outs 取出的值无法放到outputs中正确的位置。
我的代码中改为 top = row_ind*self.anchor_num+j*length 是有效的。
Lines 86 to 91 in f865b7c
| top = row_ind+j*length | |
| left = 4*j | |
| outputs[top:top + length, 0:2] = (outs[row_ind:row_ind + length, left:left+2] * 2. - 0.5 + grid) * int(self.stride[i]) | |
| outputs[top:top + length, 2:4] = (outs[row_ind:row_ind + length, left+2:left+4] * 2) ** 2 * np.repeat(self.anchors[i, j, :].reshape(1,-1), h * w, axis=0) | |
| outputs[top:top + length, 4] = outs[row_ind:row_ind + length, 4*self.anchor_num+j] | |
| outputs[top:top + length, 5:] = outs[row_ind:row_ind + length, 5*self.anchor_num:] |
Metadata
Metadata
Assignees
Labels
No labels