基于改進(jìn)YOLOv5的火車連接鉤舌識別方法

打開文本圖片集

摘 要：為了準(zhǔn)確識別不同類型鉤舌，確保自動復(fù)鉤機(jī)器人能夠根據(jù)火車連接鉤舌狀態(tài)實(shí)時調(diào)整機(jī)械臂的位姿，提出了一種基于改進(jìn)YOLOv5的火車連接鉤舌識別方法。首先，將YOLOv5主干網(wǎng)絡(luò)中原有的C3模塊替換為梯度流豐富的C2F模塊（cross feature module），YOLOv5頸部網(wǎng)絡(luò)中原有的C3模塊替換為基于FasterNet模塊構(gòu)建的輕量化C3_FasterNet模塊，并將CoordConv模塊嵌入到Y(jié)OLOv5的主干網(wǎng)絡(luò)末端。（剩余14057字）