import os os.chdir(r"/content/Mask_RCNN/samples") import sys import random import math import numpy as np import skimage.io import matplotlib import matplotlib.pyplot as plt
ROOT_DIR = os.path.abspath("../")
sys.path.append(ROOT_DIR) # To find local version of the library from mrcnn import utils from mrcnn import model as modellib from mrcnn import visualize
sys.path.append(os.path.join(ROOT_DIR, "samples/coco/")) # To find local version import coco
%matplotlib inline
MODEL_DIR = os.path.join(ROOT_DIR, "logs")
COCO_MODEL_PATH = os.path.join(ROOT_DIR, "mask_rcnn_coco.h5")
if not os.path.exists(COCO_MODEL_PATH): utils.download_trained_weights(COCO_MODEL_PATH)
IMAGE_DIR = os.path.join(ROOT_DIR, "images")