From f1749b0b03690c7da5773292e6a43d97917b2d1e Mon Sep 17 00:00:00 2001 From: Rigel-1994 Date: Fri, 6 Jul 2018 11:15:24 +0800 Subject: [PATCH] modify crop coordinates --- pytorch/src/pre_process.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pytorch/src/pre_process.py b/pytorch/src/pre_process.py index 46b86e6..54aed47 100644 --- a/pytorch/src/pre_process.py +++ b/pytorch/src/pre_process.py @@ -71,8 +71,8 @@ def image_test(resize_size=256, crop_size=224): std=[0.229, 0.224, 0.225]) #ten crops for image when validation, input the data_transforms dictionary start_first = 0 - start_center = (resize_size - crop_size - 1) / 2 - start_last = resize_size - crop_size - 1 + start_center = (resize_size - crop_size) / 2 + start_last = resize_size - crop_size return transforms.Compose([ ResizeImage(resize_size), @@ -86,8 +86,8 @@ def image_test_10crop(resize_size=256, crop_size=224): std=[0.229, 0.224, 0.225]) #ten crops for image when validation, input the data_transforms dictionary start_first = 0 - start_center = (resize_size - crop_size - 1) / 2 - start_last = resize_size - crop_size - 1 + start_center = (resize_size - crop_size) / 2 + start_last = resize_size - crop_size data_transforms = {} data_transforms['val0'] = transforms.Compose([ ResizeImage(resize_size),ForceFlip(),