From 840bca878eeb294463b0ba1c0646ea3148b6cf01 Mon Sep 17 00:00:00 2001 From: Chunk Date: Wed, 22 Apr 2015 10:49:17 +0800 Subject: [PATCH] find . -name '*.py' -o -name *.sh | xargs wc -l \n 11305 total. --- mdata/ILSVRC.py | 3 ++- test/test_model.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/mdata/ILSVRC.py b/mdata/ILSVRC.py index 2146568..36ac1db 100644 --- a/mdata/ILSVRC.py +++ b/mdata/ILSVRC.py @@ -458,7 +458,8 @@ class DataILSVRC(DataDumperBase): for tag, feat in dict_dataset.values(): feat.ravel()[[i*304+j for i in range(0,304,8) for j in range(0,304,8)]] = 0 - X.append(feat.ravel()) + feat = np.bitwise_and(feat,1) + X.append(feat.ravel()) Y.append(int(tag)) else: diff --git a/test/test_model.py b/test/test_model.py index 1d633a2..9136f87 100755 --- a/test/test_model.py +++ b/test/test_model.py @@ -158,6 +158,7 @@ def test_THEANO_crop(): timer.mark() dilc = ILSVRC.DataILSVRC(base_dir='/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val', category='Test_crop_pil') X, Y = dilc.load_data(mode='local', feattype='coef') + print X[0],Y timer.report() # X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0) -- libgit2 0.21.2