인공지능/yolov5 (crack detection)
-
yolov5 train custom data인공지능/yolov5 (crack detection) 2021. 11. 3. 15:21
crack detection에 대한 과제로 defectnet에 대한 논문을 봤지만 데이터셋도 비공개이고 github에 있는 코드도 잘 안돌아가서(?) yolov5로 crack detection 하는 걸로 정해졌다. yolov5 train custom data dataset은 Kaggle에 있는 concreteCrackSegmentationDataset + crack_segmentation_dataset 으로 했다. 공개 dataset의 양이 적어 augmentation data를 추가하기로 했다. 기존의 dataset(Kaggle data)에 이미지블렌딩을 해서 dataset의 양을 키웠다. (20만장정도 dataset에 추가했다.) train 진행 1. Train : Ag Data(20만장) + Rea..