Skip to content

Instantly share code, notes, and snippets.

@bhavesh-b
Created August 3, 2020 16:01
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bhavesh-b/029deda78789b9d3ca10690b9833274d to your computer and use it in GitHub Desktop.
Save bhavesh-b/029deda78789b9d3ca10690b9833274d to your computer and use it in GitHub Desktop.
# Some basic setup:
# Setup detectron2 logger
import detectron2
from detectron2.utils.logger import setup_logger
setup_logger()
# import some common libraries
import numpy as np
import os, json, cv2, random
from google.colab.patches import cv2_imshow
# import some common detectron2 utilities
from detectron2 import model_zoo
from detectron2.engine import DefaultPredictor
from detectron2.config import get_cfg
from detectron2.utils.visualizer import Visualizer
from detectron2.data import MetadataCatalog, DatasetCatalog
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment