Detectron.pytorch代码解读--tools test_net.py

 

代码

这个程序的主要目的是 ："""Perform inference on one or more datasets."""

在一个或多个测试集上进行测试

最后返回结果包括  预测的结果文件.json

detections.pkl

分析结果 map recall等（当然是调用其他函数实现的）

本程序的代码功能是解析输入的参数 等

主要的函数在于

    run_inference(
        args,
        ind_range=args.range,
        multi_gpu_testing=args.multi_gpu_testing,
        check_expected_results=True)

 

而 run_inference 来自下面这个文件所以

from core.test_engine import run_inference

下次分析  core.test_engine

    run_inference(
        args,
        ind_range=args.range,
        multi_gpu_testing=args.multi_gpu_testing,
        check_expected_results=True)

"""Perform inference on one or more datasets."""

import argparse
import cv2
import os