From dd62d6fa4a85f18a29de009a972f5599b19ec946 Mon Sep 17 00:00:00 2001 From: natanielruiz <nataniel777@hotmail.com> Date: 星期四, 14 九月 2017 00:51:53 +0800 Subject: [PATCH] Fixing hopenet --- code/test_on_video.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/code/test_on_video.py b/code/test_on_video.py index 8d6c5fd..d384b08 100644 --- a/code/test_on_video.py +++ b/code/test_on_video.py @@ -60,7 +60,7 @@ print 'Loading data.' transformations = transforms.Compose([transforms.Scale(224), - transforms.RandomCrop(224), transforms.ToTensor(), + transforms.CenterCrop(224), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) model.cuda(gpu) -- Gitblit v1.8.0