CDSet: 開源斑馬線檢測(cè)數(shù)據(jù)集和相應(yīng)開源代碼

CDSet數(shù)據(jù)集(Crosswalk DataSet)是在車載攝像頭上采集到的斑馬線圖像,用于研究先進(jìn)的斑馬線目標(biāo)檢測(cè)和汽車過線行為分析的神經(jīng)網(wǎng)絡(luò)算法。

CDSet數(shù)據(jù)集包含3434張圖像,包括白天、雨天、遮擋、變形、截?cái)唷⒁雇?、破損、炫光等真實(shí)場(chǎng)景,目標(biāo)包含斑馬線和導(dǎo)流線兩個(gè)類別,數(shù)據(jù)集被切分為3080張訓(xùn)練集和354張測(cè)試集。此外,本數(shù)據(jù)集還提供了額外的標(biāo)注為有無斑馬線的1770張圖像用于測(cè)試。
對(duì)應(yīng)論文`CDNet: a real-time and robust crosswalk detection network on Jetson nano based on YOLOv5`,點(diǎn)此閱讀論文:https://rdcu.be/cHuc8, DOI: 10.1007/s00521-022-07007-9。CDNet基于當(dāng)時(shí)的SOTA模型YOLOv5并進(jìn)行了ROI、SE、NST、SSVM、Synthetic Fog等多項(xiàng)改進(jìn),在Nvidia Jetson Nano上實(shí)現(xiàn)了33.1FPS和94.83% F1分?jǐn)?shù)的實(shí)時(shí)和魯棒效果。開源代碼見https://github.com/zhangzhengde0225/CDNet.
數(shù)據(jù)集和代碼均可免費(fèi)用于學(xué)術(shù)研究。