arXiv reaDer
エッジ検出の一般化のための小型で効率的なモデル
Tiny and Efficient Model for the Edge Detection Generalization
ほとんどの高レベルのコンピューター ビジョン タスクは、初期プロセスとして低レベルの画像操作に依存します。エッジ検出、画像強調、超解像度などの操作は、より高度な画像分析の基礎を提供します。現在の最先端 (SOTA) エッジ検出モデルは精度を高めるために複雑さが増しているため、この作業では、シンプルさ、効率性、一般化という 3 つの主な目的を考慮してエッジ検出に取り組みます。これを達成するために、最先端モデルの 0.2% 未満である 58,000 個のパラメーターのみを備えた軽量畳み込みニューラル ネットワークである Tiny and Efficient Edge Detector (TEED) を紹介します。 BIPED データセットでのトレーニングには 30 分未満かかり、各エポックにかかる時間は 5 分未満です。私たちが提案したモデルはトレーニングが簡単で、最初の数エポック内ですぐに収束しますが、予測されたエッジマップは鮮明で高品質です。さらに、エッジ検出と画像セグメンテーションで使用される一般的な画像からのサンプルで構成される、エッジ検出の一般化をテストするための新しいデータセットを提案します。ソース コードは https://github.com/xavysp/TEED で入手できます。
Most high-level computer vision tasks rely on low-level image operations as their initial processes. Operations such as edge detection, image enhancement, and super-resolution, provide the foundations for higher level image analysis. In this work we address the edge detection considering three main objectives: simplicity, efficiency, and generalization since current state-of-the-art (SOTA) edge detection models are increased in complexity for better accuracy. To achieve this, we present Tiny and Efficient Edge Detector (TEED), a light convolutional neural network with only 58K parameters, less than 0.2% of the state-of-the-art models. Training on the BIPED dataset takes less than 30 minutes, with each epoch requiring less than 5 minutes. Our proposed model is easy to train and it quickly converges within very first few epochs, while the predicted edge-maps are crisp and of high quality. Additionally, we propose a new dataset to test the generalization of edge detection, which comprises samples from popular images used in edge detection and image segmentation. The source code is available in https://github.com/xavysp/TEED.
updated: Sat Aug 12 2023 05:23:36 GMT+0000 (UTC)
published: Sat Aug 12 2023 05:23:36 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アソシエイト