OpenMMLab AI 实战训练营笔记 03

PyTorch简介

OpenMMLab AI 实战训练营笔记 03_第1张图片

PyTorch的基本模块

OpenMMLab AI 实战训练营笔记 03_第2张图片
OpenMMLab AI 实战训练营笔记 03_第3张图片
OpenMMLab AI 实战训练营笔记 03_第4张图片
OpenMMLab AI 实战训练营笔记 03_第5张图片
OpenMMLab AI 实战训练营笔记 03_第6张图片
OpenMMLab AI 实战训练营笔记 03_第7张图片

MMClassification

OpenMMLab AI 实战训练营笔记 03_第8张图片

MMClassification的项目结构

OpenMMLab AI 实战训练营笔记 03_第9张图片

图像分类模型构成

OpenMMLab AI 实战训练营笔记 03_第10张图片

config 构建图像分类模型

OpenMMLab AI 实战训练营笔记 03_第11张图片
OpenMMLab AI 实战训练营笔记 03_第12张图片
OpenMMLab AI 实战训练营笔记 03_第13张图片
OpenMMLab AI 实战训练营笔记 03_第14张图片

推理API

from mmcls.apis import init_model, inference_model

model = init_model('mobilenet-v2_8xb32_in1k.py', 
                   'mobilenet_v2_batch256_imagenet_20200708-3b2dc3af.pth', 
                   device='cuda:0')
result = inference_model(model, 'banana.png')

from mmcls.apis import show_result_pyplot

show_result_pyplot(model, 'banana.png', result)

OpenMMLab AI 实战训练营笔记 03_第15张图片

你可能感兴趣的:(人工智能,深度学习,pytorch)