Skip to content
Snippets Groups Projects
Commit 074f83e9 authored by Ross Girshick's avatar Ross Girshick
Browse files

use flipped examples for SVM training, too

parent 209643b4
No related branches found
No related tags found
No related merge requests found
......@@ -292,8 +292,15 @@ if __name__ == '__main__':
imdb_train = get_imdb(args.imdb_name)
print 'Loaded dataset `{:s}` for training'.format(imdb_train.name)
# enhance roidb to contain flipped examples
if cfg.TRAIN.USE_FLIPPED:
print 'Appending horizontally-flipped training examples...'
imdb_train.append_flipped_roidb()
print 'done'
trainer = SVMTrainer(net, imdb_train)
trainer.train()
filename = '{}/{}.caffemodel'.format(out_dir, out)
net.save(filename)
print 'Wrote svm model to: {:s}'.format(filename)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment