mmdetection 修改预训练模型权重类别数
修改预训练权重类别数import osimport torchimport argparsedef init_args(): parser = argparse.ArgumentParser() parser.add_argument("--org_path", type=str, help="the path of pretrained model") parse...
原创
2019-12-11 14:10:48 ·
2433 阅读 ·
6 评论