3e831835a078c678e6d293cf8c64c2761c8dbb0d,examples/fpn/demo.py,,main,#,17

Before Change

    if args.model == "faster_rcnn_fpn_resnet50":
        mode = "bbox"
        model = FasterRCNNFPNResNet50(
            n_fg_class=len(coco_bbox_label_names),
            pretrained_model=args.pretrained_model)
    elif args.model == "faster_rcnn_fpn_resnet101":
        mode = "bbox"
        model = FasterRCNNFPNResNet101(
            n_fg_class=len(coco_bbox_label_names),
            pretrained_model=args.pretrained_model)
    elif args.model == "mask_rcnn_fpn_resnet50":
        mode = "instance_segmentation"
        model = MaskRCNNFPNResNet50(
            n_fg_class=len(coco_instance_segmentation_label_names),

After Change

        default="faster_rcnn_fpn_resnet50")
    parser.add_argument("--gpu", type=int, default=-1)
    parser.add_argument("--pretrained-model")
    parser.add_argument(
        "--dataset", choices=("coco"), default="coco")
    parser.add_argument("image")
    args = parser.parse_args()

    if args.model == "faster_rcnn_fpn_resnet50":

Instances