- 博客(5)
- 收藏
- 关注
RSS订阅原创 使用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
4597
2
原创 用pytorch实现tensorflow卷积中的SAME padding
class SamePad2d(nn.Module): """Mimics tensorflow's 'SAME' padding. """ def __init__(self, kernel_size, stride): super(SamePad2d, self).__init__() self.kernel_size = torch....
2019-09-11 21:48:08
1903
原创 使用tensorflow API中的 object-detection ImportError: cannot import name 'string_int_label_map_pb2'
在使用tensorflow API中的物体检测时，报错为没有名为string_int_label_map_pb2.py的文件
2019-09-11 09:41:28
2255
原创 torch.load()加载模型时报错
torch.load()加载模型时出现如下错误Traceback (most recent call last): File "demo_syncnet.py", line 26, in <module> s.loadParameters(opt.initial_model); File "/media/cj/75bb371d-0b6d-4995-bb72-060a4...
2019-09-10 20:01:36
11443
7
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人