代码链接:
https://github.com/iduta/iresnet
1070 15ms
6, 3, 512, 512 100ms
if __name__ == '__main__':
model=iresnet50(pretrained=False).cuda()
model.eval()
x = torch.randn(1, 3, 512, 512).cuda()
# x = torch.randn(64,224,7,7).cuda()
for i in range(10):
start = time.time()
out = model(x)
print(time.time() - start, out.size())
IResNet
参考: