AI开发平台MODELARTS-本地交互式推理:VisPlatform

时间：2024-05-11 08:28:52

AI开发平台MODELARTS

from modelarts.algo_kits import VisDetPlatform

VisDetPlatform(learner, stage="val", det_box_color=(0, 0, 255))

表1 Valplatform参数说明
参数名称	可选/必选	参数类型	参数描述
learner	可选	Learner object	Learner对象，用于在线实时推理，在离线推理或数据集可视化场景可省略。
stage	可选	string	推理图片所属的DataBlock数据集类别，包括"train"、"val"、"test"，默认为"train"，只对在线推理有效。
classes	可选	list	需要展示的类别列表。
ann_json	可选	string	标注文件路径，主要用于离线展示。
det_file	可选	string	验证输出的检测结果文件路径，主要用于离线展示。
det_box_color	可选	tuple	检测框颜色，默认为(0, 0, 255)
gt_box_color	可选	string	GT标注框颜色，默认为(80, 127, 255)
without_gt	可选	bool	离线推理是否包含gt信息。
mask_palette	可选	list	mask蒙版颜色列表，如果不设置，则使用随机颜色，只对mask类算法生效。
infer_param	可选	dict	推理相关参数，如不指定则使用配置文件默认参数，与推理速度相关，目前只对mmdetection类模型生效，主要包含： 1. nms_pre：nms操作前生成的bbox数量 2. max_per_img：每张图片最大目标数 3. img_scale：图像resize尺寸。