This is an automated email from the ASF dual-hosted git repository. sxjscience pushed a commit to branch vision in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/vision by this push: new e9ae848 Revert "add test script" e9ae848 is described below commit e9ae848167a46d4a31cbb36e50fecbce36c688bc 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>'].