Configuration for training on MS COCO. Derives from the base Config class and overrides values specific to the COCO dataset.
| 66 | |
| 67 | |
| 68 | class CocoConfig(Config): |
| 69 | """Configuration for training on MS COCO. |
| 70 | Derives from the base Config class and overrides values specific |
| 71 | to the COCO dataset. |
| 72 | """ |
| 73 | # Give the configuration a recognizable name |
| 74 | NAME = "coco" |
| 75 | |
| 76 | # We use a GPU with 12GB memory, which can fit two images. |
| 77 | # Adjust down if you use a smaller GPU. |
| 78 | IMAGES_PER_GPU = 2 |
| 79 | |
| 80 | # Uncomment to train on 8 GPUs (default is 1) |
| 81 | # GPU_COUNT = 8 |
| 82 | |
| 83 | # Number of classes (including background) |
| 84 | NUM_CLASSES = 1 + 80 # COCO has 80 classes |
| 85 | |
| 86 | |
| 87 | ############################################################ |