石油泄漏
对抗制
图像(数学)
人工智能
计算机科学
环境科学
计算机视觉
环境工程
作者
Yue-Peng Cai,Lusheng Chen,Xuebin Zhuang,Bolin Zhang
标识
DOI:10.1016/j.marpolbul.2024.116475
摘要
As marine resources and transportation develop, oil spill incidents are increasing, endangering marine ecosystems and human lives. Rapidly and accurately identifying marine oil spill is of utmost importance in protecting marine ecosystems. Marine oil spill detection methods based on deep learning and computer vision have the great potential significantly enhance detection efficiency and accuracy, but their performance is often limited by the scarcity of real oil spill samples, posing a challenging to train a precise detection model. This study introduces a detection method specifically designed for scenarios with limited sample sizes. First, the small sample dataset of marine oil spill taken by Landsat-8 satellite is used as the training set. Then, a single image generative adversarial network (SinGAN) capable of training with a single oil spill image is constructed for expanding the dataset, generating diverse marine oil spill samples with different shapes. Second, a YOLO-v8 model is pretrained via the method of transfer learning and then trained with dataset before and after augmentation separately for real-time and efficient oil spill detection. Experimental results have demonstrated that the YOLO-v8 model, trained on an expanded dataset, exhibits notable enhancements in recall, precision, and average precision, with improvements of 12.3 %, 6.3 %, and 11.3 % respectively, compared to the unexpanded dataset. It reveals that our marine oil spill detection model based on YOLO-v8 exhibits leading or comparable performance in terms of recall, precision, and AP metrics. The data augmentation technique based on SinGAN contributes to the performance of other popular object detection algorithms as well.
科研通智能强力驱动
Strongly Powered by AbleSci AI