yolo和yolo-tiny實際使用對比

最近在學習使用yolo進行目標識別,然而由於顯存問題,yolo官方的train.py跑不了,所以嘗試了兩種方法,一種是減小yolo訓練時的batch_size,另一種是使用yolo-tiny,現記錄一下兩者的效果對比。 先交代一下前提,我的顯卡是GTX1050Ti的,顯存4G,訓練只有一種類別,訓練集有140張圖,平均每張圖上有3個左右目標。使用的是yolov3。 首先使用yolo-tiny,ba
相關文章
相關標籤/搜索