arXiv reaDer
LeafGAN:実用的な植物病害診断のための効果的なデータ増強方法
LeafGAN: An Effective Data Augmentation Method for Practical Plant Disease Diagnosis
植物病害の自動診断のための多くのアプリケーションは、深層学習技術の成功に基づいて開発されてきました。ただし、これらのアプリケーションは過剰適合に悩まされることが多く、新しい環境のテストデータセットで使用すると、診断パフォーマンスが大幅に低下します。本論文では、独自の注意メカニズムを備えた新しい画像間変換システムであるLeafGANを提案する。 LeafGANは、植物の病気の診断のパフォーマンスを向上させるためのデータ拡張ツールとして、健康な画像からの変換を介してさまざまな病気の画像を生成します。独自の注意メカニズムのおかげで、私たちのモデルは、さまざまな背景を持つ画像から関連する領域のみを変換できるため、トレーニング画像の多様性が高まります。 5クラスのキュウリ病分類を使用した実験では、バニラCycleGANを使用したデータ拡張は、一般化の改善に役立たないことが示されています。つまり、病気の診断パフォーマンスはベースラインからわずか0.7%増加しました。対照的に、LeafGANは診断パフォーマンスを7.4%向上させました。また、LeafGANによって生成された画像は、バニラCycleGANによって生成された画像よりもはるかに高品質で説得力があることを視覚的に確認しました。コードはhttps://github.com/IyatomiLab/LeafGANで公開されています。
Many applications for the automated diagnosis of plant disease have been developed based on the success of deep learning techniques. However, these applications often suffer from overfitting, and the diagnostic performance is drastically decreased when used on test datasets from new environments. In this paper, we propose LeafGAN, a novel image-to-image translation system with own attention mechanism. LeafGAN generates a wide variety of diseased images via transformation from healthy images, as a data augmentation tool for improving the performance of plant disease diagnosis. Thanks to its own attention mechanism, our model can transform only relevant areas from images with a variety of backgrounds, thus enriching the versatility of the training images. Experiments with five-class cucumber disease classification show that data augmentation with vanilla CycleGAN cannot help to improve the generalization, i.e., disease diagnostic performance increased by only 0.7% from the baseline. In contrast, LeafGAN boosted the diagnostic performance by 7.4%. We also visually confirmed the generated images by our LeafGAN were much better quality and more convincing than those generated by vanilla CycleGAN. The code is available publicly at: https://github.com/IyatomiLab/LeafGAN.
updated: Fri Nov 27 2020 13:34:44 GMT+0000 (UTC)
published: Mon Feb 24 2020 07:36:56 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト