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

how the sigmas will be used in testing

harlem867 opened this issue · comments

@motokimura hello, motokimura, thank you for this great project!
In the paper of gaussian yolov3, the mean of sigma_x,sigma_y,sigma_w,sigma_h will be used to calculate the uncertainty, and (1-uncertainty) will be multiplied with obj_conf and class_conf.

How do you use the sigmas while testing in this code?
I didn't find in the function of postprocess. Thank you!

Hi @harlem867!
(1 - uncertainty) is multiplied to obj_conf at these lines.
After this operation, pred is input to postprocess() function.