From 628441b7e4a6d627cdf301581dbd3e0e718455e6 Mon Sep 17 00:00:00 2001
From: ThangVu <thangvubk@gmail.com>
Date: Thu, 20 Dec 2018 08:33:10 +0000
Subject: [PATCH] caffe2 preprocess in group norm unit test
---
tools/train_imagenet/train_imagenet.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tools/train_imagenet/train_imagenet.py b/tools/train_imagenet/train_imagenet.py
index 0a9b367..202344d 100644
--- a/tools/train_imagenet/train_imagenet.py
+++ b/tools/train_imagenet/train_imagenet.py
@@ -197,8 +197,7 @@ def main_worker(gpu, ngpus_per_node, args):
traindir = os.path.join(args.data, 'train')
valdir = os.path.join(args.data, 'val')
normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
- std=[0.229, 0.224, 0.225])
-
+ std=[1/255, 1/255, 1/255])
train_dataset = datasets.ImageFolder(
traindir,
transforms.Compose([
@@ -321,6 +320,7 @@ def validate(val_loader, model, criterion, args):
if args.gpu is not None:
input = input.cuda(args.gpu, non_blocking=True)
target = target.cuda(args.gpu, non_blocking=True)
+ input = torch.cat([input[:, 2:3, :, :], input[:, 1:2, :, :], input[:, 0:1, :, :]], dim=1)
# compute output
output = model(input)
--
GitLab