motokimura / PyTorch_Gaussian_YOLOv3

PyTorch implementation of Gaussian YOLOv3 (including training code for COCO dataset)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Does this sentence mean total uncertainty? thanks for the reply! ! pred[..., 4] *= (1.0 - sigma)

xiaozhangya opened this issue · comments

Does this sentence mean total uncertainty? thanks for the reply! ! pred[..., 4] *= (1.0 - sigma)

That equation corresponds to eq. 10 of the original paper.

sigma is the mean of the uncertainties for x, y, w, and h coords
(sigma shape is [batch, anchor x grid_y x grid_x]).

pred[..., 4] is object confidences.