This is an automated email from the ASF dual-hosted git repository.
jxie pushed a commit to branch vision
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 400c380881e98291c47055ceaf89933a3debaa72
Author: Xingjian Shi <xsh...@ust.hk>
AuthorDate: Tue Nov 21 11:08:23 2017 -0800
Revert "add test script"
This reverts commit 23f68272e305103ad87d089e700ef715b13067c0.
---
test_new_image_loader.py | 34 ----------------------------------
1 file changed, 34 deletions(-)
diff --git a/test_new_image_loader.py b/test_new_image_loader.py
deleted file mode 100644
index 296869e..0000000
--- a/test_new_image_loader.py
+++ /dev/null
@@ -1,34 +0,0 @@
-import os
-os.environ['MXNET_CPU_WORKER_NTHREADS'] = '1'
-os.environ['OMP_NUM_THREADS'] = '1'
-import time
-import numpy as np
-import multiprocessing as mp
-import mxnet as mx
-from mxnet import gluon as gl
-from mxnet.gluon.data.vision import transforms
-
-if __name__ == '__main__':
- M = 24
- BS = 100
-
- dataset = gl.data.vision.ImageFolderDataset('../256_ObjectCategories')
- transform = transforms.Compose([transforms.ToTensor(),
-
transforms.RandomBrightness(1.0),
-
transforms.RandomContrast(1.0),
-
transforms.RandomSaturation(1.0),
-
transforms.Normalize([0, 0, 0], [1, 1, 1])])
- dataset = dataset.transform_first(lambda x:
transform(mx.image.center_crop(x, (224, 224))[0]))
- data_loader = gl.data.DataLoader(dataset, BS, shuffle=True,
num_workers=M)
-
- N = len(dataset)
-
- iterator = iter(data_loader)
-
- tic = time.time()
-
- for data, label in iterator:
- data.wait_to_read()
- print(data.shape)
-
- print(N/(time.time() - tic))
--
To stop receiving notification emails like this one, please contact
"comm...@mxnet.apache.org" <comm...@mxnet.apache.org>.
