diff --git a/train_frcnn.py b/train_frcnn.py index ee5b4990..dffa7d26 100644 --- a/train_frcnn.py +++ b/train_frcnn.py @@ -109,7 +109,7 @@ #train_imgs = [s for s in all_imgs if s['imageset'] == 'trainval'] #val_imgs = [s for s in all_imgs if s['imageset'] == 'test'] -print(f'Num train samples {len(train_imgs}') +print(f'Num train samples {len(train_imgs)}') print(f'Num val samples {len(val_imgs)}')