arXiv reaDer
修正されたFSSDとモデル圧縮に基づく小さなオブジェクトの検出
Small Object Detection Based on Modified FSSD and Model Compression
小さな物体は解像度が比較的低く、抽出が困難な不明瞭な視覚的特徴があるため、既存の物体検出方法では小さな物体を効果的に検出できず、検出速度と安定性が劣ります。そこで、本論文では、FSSDに基づく小物体検出アルゴリズムを提案する一方で、計算コストとストレージスペースを削減するために、モデルの圧縮を実現するためにプルーニングを実行します。まず、さまざまなレイヤーの特徴に含まれるセマンティック情報を使用して、さまざまなスケールオブジェクトを検出できます。また、特徴融合方法が改善され、小さなオブジェクトに有益なより多くの情報を取得できます。次に、ニューラルネットワークのトレーニングを加速し、モデルをスパースにするために、バッチ正規化レイヤーが導入されます。最後に、モデルはスケーリング係数によって剪定され、対応する圧縮モデルが取得されます。実験結果は、アルゴリズムの平均精度(mAP)がPASCAL VOCで80.4%に達する可能性があり、速度がGTX1080tiで59.5FPSであることを示しています。剪定後、圧縮モデルは79.9%mAP、79.5FPSの検出速度に達する可能性があります。 MS COCOでは、最高の検出精度(AP)は12.1%であり、IoUが0.5の場合、全体的な検出精度は49.8%APです。このアルゴリズムは、小さな物体の検出精度を向上させるだけでなく、検出速度を大幅に向上させ、速度と精度のバランスを取ります。
Small objects have relatively low resolution, the unobvious visual features which are difficult to be extracted, so the existing object detection methods cannot effectively detect small objects, and the detection speed and stability are poor. Thus, this paper proposes a small object detection algorithm based on FSSD, meanwhile, in order to reduce the computational cost and storage space, pruning is carried out to achieve model compression. Firstly, the semantic information contained in the features of different layers can be used to detect different scale objects, and the feature fusion method is improved to obtain more information beneficial to small objects; secondly, batch normalization layer is introduced to accelerate the training of neural network and make the model sparse; finally, the model is pruned by scaling factor to get the corresponding compressed model. The experimental results show that the average accuracy (mAP) of the algorithm can reach 80.4% on PASCAL VOC and the speed is 59.5 FPS on GTX1080ti. After pruning, the compressed model can reach 79.9% mAP, and 79.5 FPS in detection speed. On MS COCO, the best detection accuracy (APs) is 12.1%, and the overall detection accuracy is 49.8% AP when IoU is 0.5. The algorithm can not only improve the detection accuracy of small objects, but also greatly improves the detection speed, which reaches a balance between speed and accuracy.
updated: Tue Aug 24 2021 03:20:32 GMT+0000 (UTC)
published: Tue Aug 24 2021 03:20:32 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アソシエイト