Detectron2测试

Detectron2测试

# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
import argparse
import glob
import multiprocessing as mp
import os
import time
import cv2
import tqdm

from detectron2.config import get_cfg
from detectron2.data.detection_utils import read_image
from detectron2.utils.logger import setup_logger

from predictor import VisualizationDemo


# constants
WINDOW_NAME = "COCO detections"

confidence_threshold=0.5
opts=['MODEL.WEIGHTS', 'model_final_c10459X3.pkl']

def setup_cfg():
    # load config from file and command-line arguments
    cfg = get_cfg()
    cfg.merge_from_file("mask_rcnn_R_50_FPN_3x.yaml")
    cfg.merge_from_list(opts)
    # Set score_threshold for builtin models
    cfg.MODEL.RETINANET.SCORE_THRESH_TEST = confidence_threshold
    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = confidence_threshold
    cfg.MODEL.PANOPTIC_FPN.COMBINE.INSTANCES_CONFIDENCE_THRESH = confidence_threshold
    cfg.freeze()
    return cfg


def test1():
    setup_logger(name="fvcore")
    logger = setup_logger()
    

    cfg = setup_cfg()
    demo = VisualizationDemo(cfg)

    path="input1.jpg"
    img = read_image(path, format="BGR")
    start_time = time.time()
    predictions, visualized_output = demo.run_on_image(img)

    logger.info(
                "{}: {} in {:.2f}s".format(
                    path,
                    "detected {} instances".format(len(predictions["instances"]))
                    if "instances" in predictions
                    else "finished",
                    time.time() - start_time,
                )
            )
    

    cv2.namedWindow(WINDOW_NAME, cv2.WINDOW_NORMAL)
    cv2.imshow(WINDOW_NAME, visualized_output.get_image()[:, :, ::-1])
    cv2.imwrite("test-result1.jpg",visualized_output.get_image()[:, :, ::-1])
    
    cv2.waitKey(100)



test1()

(wind2) star@xmatrix:~/Detectron2/detectron2-master/demo$ python test-img-2020092002.py
[09/20 12:44:38 fvcore.common.checkpoint]: Loading checkpoint from model_final_c10459X3.pkl
[09/20 12:44:39 fvcore.common.checkpoint]: Reading a file from 'Detectron2 Model Zoo'
[09/20 12:44:39 fvcore.common.checkpoint]: The checkpoint state_dict contains keys that are not used by the model:
  sem_seg_head.p2.0.weight
  sem_seg_head.p2.0.norm.{weight, bias}
  sem_seg_head.p3.0.weight
  sem_seg_head.p3.0.norm.{weight, bias}
  sem_seg_head.p4.0.weight
  sem_seg_head.p4.0.norm.{weight, bias}
  sem_seg_head.p4.2.weight
  sem_seg_head.p4.2.norm.{weight, bias}
  sem_seg_head.p5.0.weight
  sem_seg_head.p5.0.norm.{weight, bias}
  sem_seg_head.p5.2.weight
  sem_seg_head.p5.2.norm.{weight, bias}
  sem_seg_head.p5.4.weight
  sem_seg_head.p5.4.norm.{weight, bias}
  sem_seg_head.predictor.{weight, bias}
[09/20 12:44:39 detectron2]: input2.jpg: detected 2 instances in 0.23s
(wind2) star@xmatrix:~/Detectron2/detectron2-master/demo$ 
(wind2) star@xmatrix:~/Detectron2/detectron2-master/demo$ 
(wind2) star@xmatrix:~/Detectron2/detectron2-master/demo$ 
(wind2) star@xmatrix:~/Detectron2/detectron2-master/demo$ 
(wind2) star@xmatrix:~/Detectron2/detectron2-master/demo$ python test-img-2020092002.py
[09/20 12:45:13 fvcore.common.checkpoint]: Loading checkpoint from model_final_c10459X3.pkl
[09/20 12:45:13 fvcore.common.checkpoint]: Reading a file from 'Detectron2 Model Zoo'
[09/20 12:45:13 fvcore.common.checkpoint]: The checkpoint state_dict contains keys that are not used by the model:
  sem_seg_head.p2.0.weight
  sem_seg_head.p2.0.norm.{weight, bias}
  sem_seg_head.p3.0.weight
  sem_seg_head.p3.0.norm.{weight, bias}
  sem_seg_head.p4.0.weight
  sem_seg_head.p4.0.norm.{weight, bias}
  sem_seg_head.p4.2.weight
  sem_seg_head.p4.2.norm.{weight, bias}
  sem_seg_head.p5.0.weight
  sem_seg_head.p5.0.norm.{weight, bias}
  sem_seg_head.p5.2.weight
  sem_seg_head.p5.2.norm.{weight, bias}
  sem_seg_head.p5.4.weight
  sem_seg_head.p5.4.norm.{weight, bias}
  sem_seg_head.predictor.{weight, bias}
[09/20 12:45:13 detectron2]: input1.jpg: detected 1 instances in 0.23s
(wind2) star@xmatrix:~/Detectron2/detectron2-master/demo$