iT邦幫忙

2022 iThome 鐵人賽

DAY 28
0
AI & Data

菜鳥工程師第一個電腦視覺(CV)專案-農作物影像辨識競賽系列 第 28

D28-競賽資料集運算Pretrained_ResNet_1st

  • 分享至 

  • xImage
  •  

Part0:前言

天氣轉冷讓人昏昏欲睡呀~~ 鐵人賽倒數幾天的文章,但所剩時間也沒多少,加上最近身體狀況不佳,好像也不能熬夜趕工了,有點力不從心Q
只能今天早點休息,明天繼續啦!


Part1:今日目標

1.下載主辦方釋出的公開測試資料集
2.EfficientNet_V2_l訓練結果
3.只使用1000個batch對預訓練ResNet-34模型進行遷移式學習


Part2:內容

1.下載主辦方釋出的公開測試資料集

順利申請到,將測試資料集的多個檔案下載,供明後天使用。

2.EfficientNet_V2_l訓練結果

昨天晚上的訓練共計10個小時,才完成一個Epoch共7162個batch的資料訓練,訓練好的模型先暫存起來,等測試資料都下載完,在對資料進行預測,並上傳到競賽平台確認模型表現。
在驗證集的準確率為49.3%,表現似乎較AlexNet好(尚未經過交叉驗證)

3.預訓練ResNet-34模型進行遷移式學習

從原先7162個batch,改為只使用1000個batch進行遷移式學習

# Download pretrained model
model_resnet34 = torchvision.models.resnet34(pretrained=True).to(device)

# Transfer Learning: (方法_2)ConvNet as fixed feature extractor
for param in model_resnet34.parameters():
    param.requires_grad = False  # 不自動更新參數:把所有層數都凍結(freeze the weights)
    
# Parameters of newly constructed modules have requires_grad=True by default
num_ftrs = model_resnet34.fc.in_features  # 最後一層輸入器的輸入維度
model_resnet34.fc = torch.nn.Linear(num_ftrs, 33)  # 33: numbers of output class

model_resnet34 = model_resnet34.to(device)
criterion = torch.nn.CrossEntropyLoss()
# Only parameters of final layer are being optimized 
optimizer_conv = torch.optim.SGD(model_resnet34.fc.parameters(), lr=0.001, momentum=0.9)  # 只對最後一層參數做更新(指定最後一層:model_conv.fc)
exp_lr_scheduler = torch.optim.lr_scheduler.StepLR(optimizer_conv, step_size=7, gamma=0.1)

### Train and evaluate
model_resnet34 = train_model(model_resnet34, criterion, optimizer_conv, exp_lr_scheduler,
                             train_batch_numbers=1000, num_epochs=1)

Part3:專案進度

完成兩個模型訓練(AlexNet和EfficientNet)。

Part4:下一步

將訓練時間縮短,並用競賽平台提供的測試資料進行不同圖片預處理的實驗,以了解有哪些方式是較有機會提升模型預測表現。


心得小語:
這個周末要好好休息睡到飽!!!
今日工時: 50min*1

如果生活把你打倒,那乾脆就直接躺著休息一下吧!
If life knocks you down, then just stay lying down.


上一篇
D27-競賽資料集運算Pretrained_EfficientNet_1st
下一篇
D29-競賽公開測試集預測表現_1st
系列文
菜鳥工程師第一個電腦視覺(CV)專案-農作物影像辨識競賽32
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言