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