| 436 | config = CocoConfig() |
| 437 | else: |
| 438 | class InferenceConfig(CocoConfig): |
| 439 | # Set batch size to 1 since we'll be running inference on |
| 440 | # one image at a time. Batch size = GPU_COUNT * IMAGES_PER_GPU |
| 441 | GPU_COUNT = 1 |
| 442 | IMAGES_PER_GPU = 1 |
| 443 | DETECTION_MIN_CONFIDENCE = 0 |
| 444 | config = InferenceConfig() |
| 445 | config.display() |
| 446 |