Skip to content

Instantly share code, notes, and snippets.

View veraposeidon's full-sized avatar
🎯
Focusing

snxo veraposeidon

🎯
Focusing
View GitHub Profile
@veraposeidon
veraposeidon / vis_model.py
Created March 14, 2019 08:00 — forked from qizhuli/vis_model.py
Visualise maskrcnn-benchmark models using pytorchviz.
# Set up custom environment before nearly anything else is imported
# NOTE: this should be the first import (no not reorder)
from maskrcnn_benchmark.utils.env import setup_environment # noqa F401 isort:skip
import argparse
import os
import torch
from maskrcnn_benchmark.config import cfg
from maskrcnn_benchmark.modeling.detector import build_detection_model