Additional comments for the previous commit

mrcnn/config.py

@@ -131,8 +131,9 @@ class Config(object):
     # the width and height, or more, even if MIN_IMAGE_DIM doesn't require it.
     # Howver, in 'square' mode, it can be overruled by IMAGE_MAX_DIM.
     IMAGE_MIN_SCALE = 0
-    # IMAGE_CHANNEL_COUNT refers to number of channels per image, which can
-    # be overridden by a subclass. RGB = 3, grayscale = 1, RGB-D = 4
+    # Number of color channels per image. RGB = 3, grayscale = 1, RGB-D = 4
+    # Changing this requires other changes in the code. See the WIKI for more
+    # details: https://github.com/matterport/Mask_RCNN/wiki
     IMAGE_CHANNEL_COUNT = 3
 
     # Image mean (RGB)
@@ -217,10 +218,10 @@ def __init__(self):
         # Input image size
         if self.IMAGE_RESIZE_MODE == "crop":
             self.IMAGE_SHAPE = np.array([self.IMAGE_MIN_DIM, self.IMAGE_MIN_DIM,
-                self.__class__.IMAGE_CHANNEL_COUNT])
+                self.IMAGE_CHANNEL_COUNT])
         else:
             self.IMAGE_SHAPE = np.array([self.IMAGE_MAX_DIM, self.IMAGE_MAX_DIM,
-                self.__class__.IMAGE_CHANNEL_COUNT])
+                self.IMAGE_CHANNEL_COUNT])
 
         # Image meta data length
         # See compose_image_meta() for details