华为云AI开发平台ModelArts构建模型_云淘科技

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
)






# 将模型加载至gpu 0上
model.to_device(0)

# 将模型加载至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参数二选一。

父主题: 通过Python API使用算法套件

同意关联代理商云淘科技,购买华为云产品更优惠(QQ 78315851)

内容没看懂? 不太想学习?想快速解决? 有偿解决: 联系专家