diff --git a/coco.py b/coco.py index ccb551b711..7ab76af322 100644 --- a/coco.py +++ b/coco.py @@ -354,6 +354,7 @@ class InferenceConfig(CocoConfig): # one image at a time. Batch size = GPU_COUNT * IMAGES_PER_GPU GPU_COUNT = 1 IMAGES_PER_GPU = 1 + DETECTION_MIN_CONFIDENCE = 0 config = InferenceConfig() config.display()