From 571b9a4592bd648d341704c8af7d457215a0e89a Mon Sep 17 00:00:00 2001 From: Eric Junyuan Xie Date: Wed, 18 Jan 2017 16:15:37 -0800 Subject: [PATCH] Update iterator.py --- example/ssd/dataset/iterator.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/example/ssd/dataset/iterator.py b/example/ssd/dataset/iterator.py index d61100281160..1a54bd03da8a 100644 --- a/example/ssd/dataset/iterator.py +++ b/example/ssd/dataset/iterator.py @@ -45,12 +45,7 @@ def __init__(self, imdb, batch_size, data_shape, \ if isinstance(data_shape, int): data_shape = (data_shape, data_shape) self._data_shape = data_shape - if isinstance(mean_pixels, list): - self._mean_pixels = mx.nd.Reshape( - mx.nd.array(mean_pixels), - shape = (3,1,1)) - else: - self._mean_pixels = mean_pixels + self._mean_pixels = mx.nd.array(mean_pixels).reshape((3,1,1)) if not rand_samplers: self._rand_samplers = [] else: