導(dǎo)入需要的包

定義網(wǎng)絡(luò)模型

構(gòu)建基本的FasterRCNN網(wǎng)絡(luò)

添加檢測新層

將新層添加到模型中

為Area of Interest (AOI) 層添加MaskRCNNPredictor，以支持識別人臉

使用Coco數(shù)據(jù)集訓(xùn)練網(wǎng)絡(luò)

定義訓(xùn)練和測試的數(shù)據(jù)集

coco數(shù)據(jù)集是2017年在ms-coco上訓(xùn)練的目標檢測數(shù)據(jù)集

train_loader

test_loader

定義優(yōu)化器，損失函數(shù)和訓(xùn)練步驟

訓(xùn)練