FunkyKoki/Look_At_Boundary_PyTorch

heatmap生成代码应该是有一个小bug

xtczq opened this issue · 2 comments

xtczq commented

train.py文件里面，(gt_heatmap[index])[(gt_heatmap[index]) < 3. * args.sigma] = \np.exp(-(gt_heatmap[index])[(gt_heatmap[index]) < 3 * args.sigma] (gt_heatmap[index])[(gt_heatmap[index]) < 3 * args.sigma] / 2. * args.sigma * args.sigma)，这个函数有问题，在2sigma*sigma少了括号

xtczq commented

dataload.py里面

FunkyKoki commented

Thank you so much for reminding me.