Commit f1aad98a84c43bb3dc9a67331c85bdd0a77419d0
1 parent
d1ad5f87
Exists in
master
and in
1 other branch
staged.
Showing
1 changed file
with
5 additions
and
4 deletions
Show diff stats
test/test_model.py
| ... | ... | @@ -151,10 +151,11 @@ def test_SVM_ILSVRC_S(): |
| 151 | 151 | |
| 152 | 152 | def test_THEANO_crop(): |
| 153 | 153 | |
| 154 | - # timer.mark() | |
| 155 | - # dilc = ILSVRC.DataILSVRC(base_dir='/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val', category='Test_crop_pil') | |
| 156 | - # X, Y = dilc.load_data(mode='local', feattype='coef') | |
| 157 | - # timer.report() | |
| 154 | + timer.mark() | |
| 155 | + dilc = ILSVRC.DataILSVRC(base_dir='/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val', category='Test_crop_pil') | |
| 156 | + X, Y = dilc.load_data(mode='local', feattype='coef') | |
| 157 | + timer.report() | |
| 158 | + | |
| 158 | 159 | # X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0) |
| 159 | 160 | # with open(os.path.join(package_dir,'../res/','ils_crop.pkl'),'wb') as f: |
| 160 | 161 | # cPickle.dump([(X_train,Y_train),(X_test,Y_test)], f) | ... | ... |