Commit 4b50bb51a1b59d67825a63da0c0dab4052b940d3
1 parent
5a469df5
Exists in
master
and in
1 other branch
staged.
Showing
1 changed file
with
8 additions
and
8 deletions
Show diff stats
test/test_model.py
| ... | ... | @@ -151,17 +151,17 @@ 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() | |
| 158 | - 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 | - cPickle.dump([(X_train,Y_train),(X_test,Y_test)], f) | |
| 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 | + # 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 | + # cPickle.dump([(X_train,Y_train),(X_test,Y_test)], f) | |
| 161 | 161 | |
| 162 | 162 | timer.mark() |
| 163 | 163 | mtheano = THEANO.ModelTHEANO(toolset='cnn') |
| 164 | - mtheano._train_cnn(dataset='../../res/ils_crop.pkl') | |
| 164 | + mtheano._train_cnn(dataset='../res/ils_crop.pkl') | |
| 165 | 165 | timer.report() |
| 166 | 166 | |
| 167 | 167 | ... | ... |