diff --git a/tools/extra/train_svms.py b/tools/extra/train_svms.py index 3b18f9a499bd7a73a0c24530e4b0661a7d07bd64..2084b750ec28c40aafad6ecb3d0b94f8bb3656ff 100755 --- a/tools/extra/train_svms.py +++ b/tools/extra/train_svms.py @@ -60,6 +60,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, :] gt_inds = np.where(roidb[i]['gt_classes'] > 0)[0] gt_boxes = roidb[i]['boxes'][gt_inds] _t.tic()