Skip to content

Commit 540ac16

Browse files
committed
Add option to disable COCO evaluation.
1 parent 7ca76b3 commit 540ac16

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

keras_maskrcnn/bin/train.py

+8-6
Original file line numberDiff line numberDiff line change
@@ -158,12 +158,14 @@ def create_generators(args):
158158
config=args.config
159159
)
160160

161-
validation_generator = CocoGenerator(
162-
args.coco_path,
163-
'val2017',
164-
batch_size=args.batch_size,
165-
config=args.config
166-
)
161+
validation_generator = None
162+
if args.evaluation:
163+
validation_generator = CocoGenerator(
164+
args.coco_path,
165+
'val2017',
166+
batch_size=args.batch_size,
167+
config=args.config
168+
)
167169
elif args.dataset_type == 'csv':
168170
from ..preprocessing.csv_generator import CSVGenerator
169171

0 commit comments

Comments
 (0)