华为云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)
|
参数名称 |
可选/必选 |
参数类型 |
参数描述 |
|---|---|---|---|
|
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）
内容没看懂？ 不太想学习？想快速解决？ 有偿解决： 联系专家
