Improved YOLOv4 for aerial object detection

Ali S., Siddique A., Ates H. F., GÜNTÜRK B. K.

29th IEEE Conference on Signal Processing and Communications Applications, SIU 2021, Virtual, Istanbul, Türkiye, 9 - 11 Haziran 2021 identifier identifier

  • Yayın Türü: Bildiri / Tam Metin Bildiri
  • Doi Numarası: 10.1109/siu53274.2021.9478027
  • Basıldığı Şehir: Virtual, Istanbul
  • Basıldığı Ülke: Türkiye
  • Anahtar Kelimeler: deep learning, object detection, small object
  • İstanbul Medipol Üniversitesi Adresli: Evet

Özet

Drones equipped with cameras are being used for surveillance purposes. These surveillance systems need vision-based object detection of ground objects which look very small because of the altitude of drones. We propose an improved YOLOv4 model targeted for vision-based small object detection. We investigated the performance of state of the art YOLOv4 object detector on the VisDrone dataset. We enhanced the features of small objects by connecting Upsampling layers and concatenating the upsampled features with the original features to obtain more refined and grained features for small objects. Experiments showed that the modified YOLOv4 achieved 2 percent better mAP results as compared to the original YOLOv4 at different image resolutions on the VisDrone dataset while running at the same speed as the original YOLOv4.