各位好,我的圖片用labelImg標好、檢查過都沒問題後,分割成以下的結構,並將此vec資料夾放到colab的yolov7/data/下面。
|_vec
|_train
| |_images
| |_labels
|_valid
|_images
|_labels
|_vec.yaml
接著我進行了以下步驟
1.打開colab,連接google drive
2.%cd /content/drive/MyDrive/yolov7
3.執行此命令
!python train.py --batch 16 --data ./data/vec/vec.yaml --cfg cfg/training/yolov7-vec.yaml --img 640 640 --epochs 50 --weights 'yolov7.pt' --name vec_epoch50 --hyp data/hyp.scratch.p5.yaml --device 0
但在train的時候,卻出現了下列訊息
train: Scanning 'data/vec/train/labels.cache' images and labels... 23 found, 82 missing, 5 empty, 0 corrupted: 100% 105/105 [00:00<?, ?it/s]
(而valid也有相同的問題)
也就是說,總共105張的圖片,他只抓到了其中23張。剩下82張的圖,他找不到label。然而我檢查過了,label們確實在該路徑下。
我用roboflow下載的資料集去跑,並沒有此問題,因此排除是v7安裝過程有任何缺失。
由於有人在v5時用Linux跑,.TXT和.txt的差異使它找不到標籤,因此我也有思考過是否式檔名的問題。但我的labels確實都是小寫的.txt。
而我的圖片有.JPG和.jpg兩種,但看上述訊息為105/105,因此我想它確實有抓到所有圖片,所以我猜測問題應該出在labels上。
但以目前的知識量,我好像無法解決這個問題,因此想請問有人遇過相同的情況嗎?該怎麼解決呢?