更新时间：2024-04-19 GMT+08:00
构建模型
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)
参数名称 |
可选/必选 |
参数类型 |
参数描述 |
|---|---|---|---|
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使用算法套件
