ckpt文件是经过训练的模型参数文件,可以根据需求选取对应的ckpt文件。一般来说,选择在经过相似任务训练的ckpt文件效果更好。如果没有类似任务的ckpt文件,则可以选择在相似数据集上训练的ckpt文件。

import cv2import numpy as npimport mindsporefrom mindspore import Tensorfrom PIL import Imagefrom main import ResNet BasicBlock# 加载模型model = mindsporetrainserializationload_checkpointckptckptnetwork =

原文地址: https://www.cveoy.top/t/topic/bqoC 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录