Arshdeep Singh 4rshdeep
wangg12
/ trim_detectron_model.py
Created
December 18, 2018 03:07
trim last layers of detectron model for maskrcnn-benchmark
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| import torch | |
| import argparse | |
| from maskrcnn_benchmark.config import cfg | |
| from maskrcnn_benchmark.utils.c2_model_loading import load_c2_format | |
| def removekey(d, listofkeys): | |
| r = dict(d) | |
| for key in listofkeys: |
OlderNewer