diff --git a/tools/extra/train_svms.py b/tools/extra/train_svms.py index 322ab52116aa54d5df631893ec97eddedd94bbb2..3b18f9a499bd7a73a0c24530e4b0661a7d07bd64 100755 --- a/tools/extra/train_svms.py +++ b/tools/extra/train_svms.py @@ -102,6 +102,8 @@ class SVMTrainer(object): # num_images = 100 for i in xrange(num_images): im = cv2.imread(self.imdb.image_path_at(i)) + if roidb[i]['flipped']: + im = im[:, ::-1, :] _t.tic() scores, boxes = im_detect(self.net, im, roidb[i]['boxes']) _t.toc()