Skip to content

Instantly share code, notes, and snippets.

@trongan93
Created April 9, 2021 05:29
Show Gist options
  • Save trongan93/29bc4baef29e620a6309f279a59b7b01 to your computer and use it in GitHub Desktop.
Save trongan93/29bc4baef29e620a6309f279a59b7b01 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 matplotlib import pyplot as plt
# 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