导入需要的包
定义网络模型
构建基本的FasterRCNN网络
添加检测新层
将新层添加到模型中
为Area of Interest (AOI) 层添加MaskRCNNPredictor,以支持识别人脸
使用Coco数据集训练网络
定义训练和测试的数据集
coco数据集是2017年在ms-coco上训练的目标检测数据集
train_loader
test_loader
定义优化器,损失函数和训练步骤