軟體工程第四次作業 貓狗大戰挑戰賽

2022-07-22 01:00:17 字數 1810 閱讀 1323

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...