From 074f83e90fb7ce2d6dfd61c21e27c80cc29f8ff2 Mon Sep 17 00:00:00 2001
From: Ross Girshick <ross.girshick@gmail.com>
Date: Fri, 17 Apr 2015 15:57:35 -0700
Subject: [PATCH] use flipped examples for SVM training, too
---
tools/extra/train_svms.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/tools/extra/train_svms.py b/tools/extra/train_svms.py
index 7e2f101..322ab52 100755
--- a/tools/extra/train_svms.py
+++ b/tools/extra/train_svms.py
@@ -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)
--
GitLab