基於Resnet50的pytorch框架下的圖像特徵提取

在Resnet50CNN結構下實現圖像的特徵提取,這裏採用的是CV2的圖像讀入方式,最後再把獲得的圖像轉換成npy格式進行輸出得得,圖像對應的特徵。網絡 # -*- coding: utf-8 -*- """ Function: 圖像特徵的提取,能夠依據需求修改CNN的輸出,獲得不一樣層網絡的輸出圖像特徵 Writer: Zenght date:2019.2.16 """ from __futu
相關文章
相關標籤/搜索