《MMClassification 介绍》课堂心得

最新推荐文章于 2024-11-08 13:43:44 发布

再多亿点可爱

最新推荐文章于 2024-11-08 13:43:44 发布

阅读量97

点赞数

文章标签： python 人工智能深度学习

本文链接：https://blog.csdn.net/m0_60082435/article/details/128876557

版权

文章介绍了如何使用MMClassification工具包进行图像分类。通过PythonAPI，可以下载配置文件和训练权重，然后在CUDA设备上进行推理。示例中，模型成功识别出banana.png图片为香蕉，预测得分接近100%。

摘要由CSDN通过智能技术生成

图像分类工具包 MMClassification

代码仓库：https://github.com/open-mmlab/mmclassification

文档教程：https://mmclassification.readthedocs.io/en/latest

Python 推理 API

# 下载配置文件和与训练权重
$ mim download mmcls - -config mobilenet - v2_8xb32_in1k - -dest .
# 使用Python API 推理
model = init_model('mobilenet-v2_8xb32_in1k.py',
                   'mobilenet_v2_batch256_imagenet_20200708-3b2dc3af.pth',
                   device='cuda:0')
result = inference_model(model, 'banana.png')
show_result_pyplot(model, 'banana.png', result)


# 推理结果result
{'pred_class': 'banana’,
 'pred_label': 954,
 'pred_score': 0.9999284744262695}