From 21e4fb3be6bfe7247dc644768f5f1db4386481b9 Mon Sep 17 00:00:00 2001
From: Bendang <bendangnuksungimsong@gmail.com>
Date: Thu, 8 Oct 2020 18:08:50 +0530
Subject: [PATCH] fix coco config

---
 config.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/config.py b/config.py
index 9028602..6f28d34 100644
--- a/config.py
+++ b/config.py
@@ -2,7 +2,7 @@
 
 
 class mask_config():
-    def __init__(self, NUMBER_OF_CLASSES):
+    def __init__(self, NUMBER_OF_CLASSES=80):
         self.NAME = "tags"
         self.IMAGES_PER_GPU = 2
         self.NUM_CLASSES = 1 + NUMBER_OF_CLASSES  # Background + tags
@@ -65,3 +65,12 @@ def __init__(self, NUMBER_OF_CLASSES):
         # Image meta data length
         # See compose_image_meta() for details
         self.IMAGE_META_SIZE = 1 + 3 + 3 + 4 + 1 + self.NUM_CLASSES
+        
+
+class CocoConfig(mask_config):
+    """Configuration for training on MS COCO.
+    Derives from the base Config class and overrides values specific
+    to the COCO dataset.
+    """
+    # Give the configuration a recognizable name
+    NAME = "coco"