2. 資料處理
datasets 是 torchvision 中的乙個包,可以用做載入影象資料。它可以以多執行緒(multi-thread)的形式從硬碟中讀取資料,使用 mini-batch 的形式,在網路訓練中向 gpu 輸送。
在使用cnn處理影象時,需要進行預處理。將被整理成
的大小,同時還將進行歸一化處理。torchvision 支援對輸入資料進行一些複雜的預處理/變換 (normalization, cropping, flipping, jittering 等)。具體可以參照 torchvision.tranforms 的官方文件說明。
3. 建立 vgg model
在這部分**中,對輸入的5個利用vgg模型進行**。
使用softmax對結果進行處理,隨後展示了識別結果。可以看到,識別結果是比較非常準確的。
4. 修改最後一層,凍結前面層的引數
把最後的 nn.linear 層由1000類,替換為2類,從而使用訓練好的模型。
設定 required_grad=false,從而在訓練中凍結前面層的引數
5. 訓練並測試全連線層
建立損失函式和優化器
訓練模型
測試模型
6. 視覺化模型**結果(主觀分析)
主觀分析就是把**的結果和相對應的測試影象輸出出來看看,一般有四種方式:
隨機檢視一些**正確的
隨機檢視一些**錯誤的
**正確,同時具有較大的probability的
**錯誤,同時具有較大的probability的
最不確定的,比如說**概率接近0.5的
有把解壓後的test挪到dogscats的test1檔案下
載入資料
測試模型
結果匯入檔案
未做任何修改,將最初結果上傳:
資料處理時,圖形縮放,再次進行檢測:
提高了訓練次數設定為3,再次進行檢測:
調整學習率為0.0001,再次進行檢測:
更改優化器sgd->adam,再次進行檢測:
努力進化中......
第四次作業 貓狗大戰挑戰賽
解讀 定義格式 normalize transforms.normalize mean 0.485,0.456,0.406 std 0.229,0.224,0.225 定義格式 vgg format transforms.compose transforms.centercrop 224 trans...
軟體工程第四次作業
部落格資訊 瀋陽航空航天大學計算機學院2020軟體工程作業 作業要求 課程目標 熟悉乙個 高質量 軟體的開發過程 作業目標 結對程式設計練習 一 題目 二 位址 三 執行結果 四 與隊友合作 工作記錄表 專案預計 實際設計時間 1h3h 編碼時間 3h5h 測試時間 30min 30min 行數 2...
軟體工程第四次作業
功能模組名稱 簡單的語法分析程式 審查人王澤鵬 審查日期 2017.4.4 名稱 黑白棋遊戲 作者 白璐檔案結構 重要性審查項 結論標頭檔案和定義檔案的名稱是否合理?合理標頭檔案和定義檔案的目錄結構是否合理?合理版權和版本宣告是否完整?不完整重要 標頭檔案是否使用了 ifndef define en...