摘要: 文章目录 前言1.导入相关包并打印版本号2.输入3.辅助函数3.1 模型训练和验证代码3.2 设置模型参数的`.requires_grad`属性 4.初始化和重塑网络4.1 Resnet4.2 Alexnet4.3 VGG4.4 Squeezenet4.5 Densenet4.6 Inception 阅读全文
posted @ 2022-02-14 18:50 小Aer 阅读(29) 评论(0) 推荐(0) 编辑
摘要: 在本教程中,我们将微调在 Penn-Fudan 数据库中对行人检测和分割的已预先训练的 Mask R-CNN模型。它包含170个图像和345个行人实例,我们 将用它来说明如何在 torchvision 中使用新功能,以便在自定义数据集上训练实例分割模型。 文章目录 1.定义数据集2.为 PennFu 阅读全文
posted @ 2022-02-14 14:13 小Aer 阅读(35) 评论(0) 推荐(0) 编辑