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,10 +151,11 @@ def test_SVM_ILSVRC_S(): | ||
| 151 | 151 | ||
| 152 | def test_THEANO_crop(): | 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 | # X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0) | 159 | # X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0) |
| 159 | # with open(os.path.join(package_dir,'../res/','ils_crop.pkl'),'wb') as f: | 160 | # with open(os.path.join(package_dir,'../res/','ils_crop.pkl'),'wb') as f: |
| 160 | # cPickle.dump([(X_train,Y_train),(X_test,Y_test)], f) | 161 | # cPickle.dump([(X_train,Y_train),(X_test,Y_test)], f) |