Unverified 提交 26c3b11f authored 作者: Francisco Ingham's avatar Francisco Ingham 提交者: GitHub

Update torch_utils.py (#751)

上级 2077d788
...@@ -159,7 +159,7 @@ def load_classifier(name='resnet101', n=2): ...@@ -159,7 +159,7 @@ def load_classifier(name='resnet101', n=2):
input_range = [0, 1] input_range = [0, 1]
mean = [0.485, 0.456, 0.406] mean = [0.485, 0.456, 0.406]
std = [0.229, 0.224, 0.225] std = [0.229, 0.224, 0.225]
for x in [input_size, input_space, input_range, mean, std]: for x in ['input_size', 'input_space', 'input_range', 'mean', 'std']:
print(x + ' =', eval(x)) print(x + ' =', eval(x))
# Reshape output to n classes # Reshape output to n classes
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论