Opened this issue 5 years ago · 1 comments
我想通过点的方式得到边界图， 在您的get_gt_heatmap里面用到了cv2的函数，我是想把通过点产生边界图的模块写成一个nn.Module. 但是不知道怎么用pytorch里面带的tensor函数去实现这个网络，您有写过这个函数吗?
就我目前所知，似乎还没有见到nn.Module中含有此类几何构建的函数，如果您现在知道了烦请告知，我也会在近期关注一下是否可以通过PyTorch完成边界图的构建。