文章介绍了如何从源代码安装MMPretrain库，包括使用pip和mim工具进行安装，并添加多模态支持。之后，通过运行python脚本来验证安装是否成功，使用了resnet18模型对JPEG图像进行预测，展示了模型在CPU或CUDA设备上的推理过程。
MMPretrain代码课
安装mmpretrain
我是选择从源码安装的。执行以下代码：
git clone https://github.com/open-mmlab/mmpretrain.git
cd mmpretrain
pip install -U openmim && mim install -e .安装多模态支持
mim install -e ".[multimodal]"验证安装
输入以下代码
python demo/image_demo.py demo/demo.JPEG resnet18_8xb32_in1k --device cpu有以下信息
或者输入以下代码
from mmpretrain import get_model, inference_model
model = get_model('resnet18_8xb32_in1k', device='cuda:0') # 或者 device='cuda:0'
inference_model(model, 'demo/demo.JPEG',)会看到输出一个字典，包含预测的标签、得分及类别名。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
