Object detection in remote sensing imagery is crucial for numerous applications such as urban planning, environmental monitoring, and disaster response. This paper investigates the effectiveness of transfer learning using YOLOv8 for object detection in remote sensing, specifically focusing on the DIOR and Ships datasets. We compare models trained from scratch with those pre-trained on the COCO dataset and subsequently fine-tuned on DIOR. Our experiments demonstrate that the pre-trained YOLOv8 model significantly outperforms the scratch-trained model across all evaluated metrics, including precision, recall, and mean Average Precision (mAP). The pre-trained model demonstrates enhanced detection capabilities and reduced misclassifications. Detailed analysis using confusion matrices highlights the model’s improved ability to distinguish between visually similar categories and detect smaller, less distinct objects. These findings underscore the advantages of leveraging transfer learning to enhance the performance of object detection models in remote sensing imagery. Our study concludes that fine-tuning pre-trained models on specific datasets leads to robust and accurate detection, offering a valuable approach for remote sensing applications.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Transfer Learning with Yolo for Object Detection in Remote Sensing

  • Ema Pandilova,
  • Marko Petrov,
  • Vlatko Spasev,
  • Ivica Dimitrovski,
  • Ivan Kitanovski

摘要

Object detection in remote sensing imagery is crucial for numerous applications such as urban planning, environmental monitoring, and disaster response. This paper investigates the effectiveness of transfer learning using YOLOv8 for object detection in remote sensing, specifically focusing on the DIOR and Ships datasets. We compare models trained from scratch with those pre-trained on the COCO dataset and subsequently fine-tuned on DIOR. Our experiments demonstrate that the pre-trained YOLOv8 model significantly outperforms the scratch-trained model across all evaluated metrics, including precision, recall, and mean Average Precision (mAP). The pre-trained model demonstrates enhanced detection capabilities and reduced misclassifications. Detailed analysis using confusion matrices highlights the model’s improved ability to distinguish between visually similar categories and detect smaller, less distinct objects. These findings underscore the advantages of leveraging transfer learning to enhance the performance of object detection models in remote sensing imagery. Our study concludes that fine-tuning pre-trained models on specific datasets leads to robust and accurate detection, offering a valuable approach for remote sensing applications.