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