PaddleDetection 模型推理测试

最新推荐文章于 2024-04-18 15:58:11 发布

烩鳗鱼

最新推荐文章于 2024-04-18 15:58:11 发布

阅读量789

点赞数

分类专栏：笔记文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/m0_57173620/article/details/121462347

版权

本文详细解析了PaddleDetection的`infer.py`流程，包括从程序入口的参数初始化，如GPU加速和版本检查，到网络结构创建、模型权重加载、数据读取、推理标准选择，以及推理过程中的前向传播、结果保存和使用VisualDL展示图片结果。

摘要由CSDN通过智能技术生成

2021SC@SDUSC
本周分析

#tools/infer.py
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import os, sys
# add python path of PadleDetection to sys.path
parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 2)))
if parent_path not in sys.path:
    sys.path.append(parent_path)

# ignore numba warning
import warnings
warnings.filterwarnings('ignore')
import glob
import numpy as np
from PIL import Image

import paddle
from paddle.distributed import ParallelEnv
from ppdet.core.workspace import load_config, merge_config, create
from ppdet.utils.check import check_gpu, check_version, check_config
from ppdet.utils.visualizer import visualize_results
from ppdet.utils.cli import ArgsParser
from ppdet.utils.checkpoint import load_weight
from ppdet.utils.eval_utils import get_infer_results

from ppdet.utils.logger import setup_logger
logger = setup_logger('train')

#运行配置参数解析