Browse Source

Update torch_utils.py (#751)

5.0
Francisco Ingham GitHub 4 years ago
parent
commit
26c3b11fdc
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      utils/torch_utils.py

+ 1
- 1
utils/torch_utils.py View File

@@ -159,7 +159,7 @@ def load_classifier(name='resnet101', n=2):
input_range = [0, 1]
mean = [0.485, 0.456, 0.406]
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))

# Reshape output to n classes

Loading…
Cancel
Save