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

你好 作者！有一些关系boundary maps 的问题想请教一下 #12

Open wqz960 opened 5 years ago

wqz960 commented 5 years ago

我想通过点的方式得到边界图， 在您的get_gt_heatmap里面用到了cv2的函数，我是想把通过点产生边界图的模块写成一个nn.Module. 但是不知道怎么用pytorch里面带的tensor函数去实现这个网络，您有写过这个函数吗?

FunkyKoki commented 4 years ago

就我目前所知，似乎还没有见到nn.Module中含有此类几何构建的函数，如果您现在知道了烦请告知，我也会在近期关注一下是否可以通过PyTorch完成边界图的构建。