由於是初次進行實作,在過程中遇到了滿多不同的問題,像一開始資料的分割以及處理等,最難受的就是套件版本的問題,當初在安裝套件時,參考了ultralytics官方的依賴項進行安裝套件,結果在訓練時,跳出了數多套件引發的錯誤,在一知半解的狀況下完全沒有頭緒,依賴項如下圖。
錯誤的資訊上出現了Pytorch不可用的狀況,但使用了torch.cuda.is_available()
得出的回應是True
,這讓筆者感到很疑惑。
由於筆者對此也沒有頭緒,於是便試著尋找ChatGPT
求助,將問題丟上去後,得出的回應卻是如下圖的流水文,即使嘗試了問題仍然依舊。
後來借助著實驗室的學長幫忙,以及上官方的github上爬文是否有人有遇到相同的問題,終於是讓筆者爬到了這篇文章,問題如下圖,與筆者發生的狀況相同。
後來有翻到官方人員對這個問題的回應,似乎是torchvision
的版本上出現了問題導致CUDA的不支援,嘗試了更新版本後,果然錯誤都被解決了。
這個問題困擾了筆者不少時間,因為一開始的問題錯誤並不是直接指向torch方面的問題,而是其他套件的錯誤,因此筆者當時只針對出現錯誤的項目去進行降版本以及爬文修正的部分,但問題仍然依舊,但後續通過這篇issue以及學長幫忙debug後,終於順利地解決。
得到解決後,模型也順利的訓練完了,後續筆者將對訓練完的模型做精準度以及對車牌實際的測試,那這邊我們下篇見。