AI开发平台MODELARTS-构建模型

时间：2024-04-19 15:33:22

构建模型

from modelarts.algo_kits import Model

model = Model(
	env,
    model_name="mmdetection:fcos/fcos_r50_caffe_fpn_gn-head_1x_coco",
    num_classes=80,
    checkpoint="your_local_path_of_pretrained_model",
    load_default_backbone=False
)

# 将模型加载至cpu上
model.to_device(-1)

表1 Model参数说明
参数名称	可选/必选	参数类型	参数描述
env	必选	EnvManager Object	上下文管理对象，初始化方法请参考通过Python API使用算法套件。
model_name	必选	string	模型名称。
num_classes	可选	int	类别数，默认会从标注信息中读取。
checkpoint	可选	string	本地预训练模型路径，默认为None，使用默认值时随机生成网络参数。
load_default_backbone	可选	boolean	是否加载默认的预训练骨干网络，如resnet50，默认为False，该参数设置为True时模型自动从open-mmlab中拉取，可与checkpoint参数二选一。