使用torchvision 中的roi_pool/roi_align函数时报错
使用torchvision 中的roi_pool函数时报错
Traceback (most recent call last):
File "/home/wkj/cj/roi/try.py", line 39, in <module>
pooled_regions = torchvision.ops.roi_align(image.cuda(),[boxes.cuda()]...
原创
2019-12-10 14:34:22 ·
4445 阅读 ·
2 评论