在PyTorch提供的已經訓練好的圖像目標檢測中,均是R-CNN系列
的網絡,并且針對目標檢測和人體關鍵點檢測分別提供了容易調用的方
法。針對目標檢測的網絡,輸入圖像均要求使用相同的預處理方式,即先將每張圖像的像素值預處理到0~1之間,且輸人的圖像尺寸不是很小即可直接
調用。已經預訓練的可供使用的網絡模型如表10-2所示。
?首先定義每個類別所對應的標簽COCO_INSTANCE_CATEGORY_NAMES,程序如下:
上面的程序在可視化圖像時,使用ImageDraw.Draw(image)方法,表示要在原始的image圖像上相應的位置添加一些元素,draw.rectangle(表示要添加矩形框,draw.text()表示在圖像上指定位置添加文本。運行程序后,可得到圖10-14所示的目標檢測結果。?
?
?