04-1 轻松学PyTorch 肺部感染识别案例简介

本期视频内容：肺部感染识别案例简介

视频地址：http://mpvideo.qpic.cn/0bc3eeabuaaa2aakaclypjrfaioddiqqagqa.f10002.mp4?dis_k=3209e3a101509e3c5e4054c7f10fdf57&dis_t=1655292607&vid=wxv_2390536430675656707&format_id=10002&support_redirect=0&mmversion=false

2.数据集展示

test:

NORMAL:235PNEUMONIA:391

train:

NORMAL:1342PNEUMONIA:3876

val:

NORMAL:9PNEUMONIA:9

正常的肺部

感染的肺部

3. 案例主要流程：

第一步：加载预训练模型ResNet，该模型已在ImageNet上训练过。

第二步：冻结预训练模型中低层卷积层的参数(权重)。

第三步：用可训练参数的多层替换分类层。

第四步：在训练集上训练分类层。

第五步：微调超参数，根据需要解冻更多层。

ResNet 网络结构图

跳连，防止梯度消失