實踐目標檢測--構造SSD網絡與訓練

網絡結構圖 基礎網絡塊可自定義,可使用VGG、ResNet、DenseNet,這裏使用了簡單三層卷積+SSD的網絡配置。 多尺度特徵塊:使用兩層3*3的網絡保持圖片大小不變後,使用2*2的平均池化,將圖片大小減半,來提取不同尺度的特徵。 類別、邊界框預測:使用多通道的輸出來預測結果。 文件名SSD_model.py import sys sys.path.insert(0, '..') impo
相關文章
相關標籤/搜索