欢迎光临散文网 会员登陆 & 注册

Pytorch , coco人脸识别卷积神经网络

2023-04-05 16:13 作者:千岛丶  | 我要投稿

导入需要的包

定义网络模型

构建基本的FasterRCNN网络

添加检测新层

将新层添加到模型中

为Area of Interest (AOI) 层添加MaskRCNNPredictor,以支持识别人脸

使用Coco数据集训练网络

定义训练和测试的数据集

coco数据集是2017年在ms-coco上训练的目标检测数据集

train_loader

test_loader

定义优化器,损失函数和训练步骤

训练


Pytorch , coco人脸识别卷积神经网络的评论 (共 条)

分享到微博请遵守国家法律