Skip to content

Commit

Permalink
also scale image during testing
Browse files Browse the repository at this point in the history
Former-commit-id: 15030d74db4e543db655d458a72865f4182b4367
  • Loading branch information
Javi Ribera committed Sep 10, 2018
1 parent 33bb408 commit 58eacaa
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion object-locator/locate.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@
try:
testset = CSVDataset(args.dataset,
transforms=transforms.Compose([
# ScaleImageAndLabel(size=(args.height, args.width)),
ScaleImageAndLabel(size=(args.height, args.width)),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5),
(0.5, 0.5, 0.5)),
Expand Down

0 comments on commit 58eacaa

Please sign in to comment.