Skip to content
 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

期中考專案總結


模型設定

  • 模型架構:ResNet-18
  • 原始影像大小:
    • 224x224 (原始版本)
    • 224x130 (改進版本)
  • 訓練目標:車道辨識,預測路徑落點
  • 平均辨識速率:60ms80ms (提提升版本)、50ms110ms (提升版本) / 150ms~200ms (原始版本)
  • 相機FPS:
    • 5fps(224x224 原始版本)
    • 9~13fps(224x130 改進版本)
  • 參數
  • image.png

訓練資料集與結果

版本名稱 描述 資料集
best_steering_model_xy.pth 第一個成功版本,使用初始的 1100 資料集進行訓練 1100
resnet18_1600.pth 新增樣本數以涵蓋更多角度 1600
resnet18_1600_v2.pth 重新標記資料以避免過於極端的標記,減少誤判 1600-v2
resnet18_1600_v3.pth 增加 100 張靠左直線空白區域樣本,增強對直線的辨識 1600-v3
resnet18_1600_v4.pth 增加 300 張靠右直線空白區域樣本,加強模型在右側車道的穩定性 1600-v4
resnet18_finetuned_fp16.pth (看起來fintuning太少) 半精度加速(量化) 1600-v4
resnet18_finetuned_fp16_pruned.pth (預測軟體看起來不錯)(以測試可以加速10% 效果不差) 半精度加速(量化)+10%剪枝 1600-v4

運行環境


改進點與優化

  • 平視拍攝
    • 拍攝角度較接近人眼平視,避免僅依賴路面特徵,有助於提前預測動向。
    • 平視需更大量樣本以忽略上半部分背景,否則易受雜訊影響。
  • 影像裁剪
    • 去除影像上半部分 40% 的資料,避免背景干擾,僅保留車道特徵,有助於提高模型穩定性。
    • 對 ResNet-18 進行輸入適配 (224x130) 訓練,顯著減少辨識延遲。
  • 半精度(量化)10%~20%剪枝
    • 將訓練好的模型減少精度FP32->FP16 加速計算
    • 並重新fintuning 減小模型.ipynb
    • 提升到60ms~80ms
  • TensorRT
    • 執行腳本(本地我搞不出來)Colab腳本

影片演示

resnet18_1600.pth 結果
resnet18_1600.pth(剪枝/半精度) 結果

最終訓練模型


YoloV4_tiny Car object detect

train.txt is form dataset

train2.txt is base on train.txt add dataset fliter by car anntoion

Main Code(Running on colab)

YoloV4.ipynb colab

Version1

  1. coco.data
  2. obj.name
  3. train.txt
  4. yolov4-tiny.weights
  5. yolov4-tiny-obj.cfg
Result

Version2

  1. coco.data
  2. obj.name
  3. train2.txt
  4. yolov4-tiny.weights
  5. yolov4-tiny-obj.cfg
Result

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages