Ensure mean subtraction happens on correct color ordering. Fixes #6

kentsommer · Apr 24, 2018 · e053637 · e053637
1 parent 7c65137
commit e053637
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/test.py b/test.py
@@ -54,6 +54,7 @@ def preprocess(images):
 	N = 0
 	mean = np.zeros((1, 3, 224, 224))
 	for X in tqdm(images_cropped):
+                X = np.transpose(X,(2,0,1)) 
 		mean[0][0] += X[:,:,0]
 		mean[0][1] += X[:,:,1]
 		mean[0][2] += X[:,:,2]
@@ -162,4 +163,4 @@ def main():
 	print 'Median error ', median_result[0], 'm  and ', median_result[1], 'degrees.'
 
 if __name__ == '__main__':
-	main()
+	main()
diff --git a/train.py b/train.py
@@ -45,6 +45,7 @@ def preprocess(images):
 	N = 0
 	mean = np.zeros((1, 3, 224, 224))
 	for X in tqdm(images_cropped):
+                X = np.transpose(X,(2,0,1))
 		mean[0][0] += X[:,:,0]
 		mean[0][1] += X[:,:,1]
 		mean[0][2] += X[:,:,2]
@@ -159,4 +160,4 @@ def main():
 
 
 if __name__ == '__main__':
-	main()
+	main()