From 75a63c4e3765580a580691e9c6a43dee31d6df5b Mon Sep 17 00:00:00 2001 From: wuzewu Date: Wed, 29 Apr 2020 20:33:58 +0800 Subject: [PATCH] Update detection demo --- demo/object_detection/train_faster_rcnn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/object_detection/train_faster_rcnn.py b/demo/object_detection/train_faster_rcnn.py index 73305a05..a2921916 100644 --- a/demo/object_detection/train_faster_rcnn.py +++ b/demo/object_detection/train_faster_rcnn.py @@ -14,7 +14,7 @@ parser = argparse.ArgumentParser(__doc__) parser.add_argument("--num_epoch", type=int, default=50, help="Number of epoches for fine-tuning.") parser.add_argument("--use_gpu", type=ast.literal_eval, default=True, help="Whether use GPU for fine-tuning.") parser.add_argument("--checkpoint_dir", type=str, default="faster_rcnn_finetune_ckpt", help="Path to save log data.") -parser.add_argument("--batch_size", type=int, default=1, help="Total examples' number in batch for training.") +parser.add_argument("--batch_size", type=int, default=5, help="Total examples' number in batch for training.") parser.add_argument("--module", type=str, default="faster_rcnn_resnet50_coco2017", help="Module used as feature extractor.") parser.add_argument("--use_data_parallel", type=ast.literal_eval, default=False, help="Whether use data parallel.") # yapf: enable. @@ -54,7 +54,7 @@ def finetune(args): config = hub.RunConfig( log_interval=10, - eval_interval=10, + eval_interval=100, use_data_parallel=args.use_data_parallel, use_pyreader=True, use_cuda=args.use_gpu, -- GitLab