voc filter

ijkguo · Aug 25, 2018 · e556bb4 · e556bb4
1 parent f8fc69e
commit e556bb4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -165,6 +165,7 @@ def get_voc(args):
     roidb = []
     for iset in isets:
         imdb = PascalVOC(iset, 'data', 'data/VOCdevkit')
+        imdb.filter_roidb()
         imdb.append_flipped_images()
         roidb.extend(imdb.roidb)
     return roidb