search

FunkyKoki / Look_At_Boundary_PyTorch

A PyTorch re-implement of CVPR 2018 LAB(Look At Bounday) --Champagne Jin
9 stars 1 forks source link

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

Open xtczq opened 5 years ago

xtczq commented 5 years ago

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)，这个函数有问题，在2sigmasigma少了括号

xtczq commented 5 years ago

dataload.py里面

FunkyKoki commented 5 years ago

Thank you so much for reminding me.