基于特征复用机制的航拍图像小目标检测算法
|
邓天民,程鑫鑫,刘金凤,张曦月
|
Small target detection algorithm for aerial images based on feature reuse mechanism
|
Tianmin DENG,Xinxin CHENG,Jinfeng LIU,Xiyue ZHANG
|
|
表 1 FS-YOLO算法的参数细节 |
Tab.1 Parameter details of FS-YOLO algorithm |
|
序号 | c1 | 模型结构 | c2 | Size | P/KB | | 序号 | c1 | 模型结构 | c2 | Size | P/KB | 1 | 3 | Conv | 32 | 320×320 | 928 | | 12 | 256 | CARAFE | 256 | 160×160 | 103 012 | 2 | 32 | FS Block_2 | 64 | 160×160 | 25 408 | 13 | 256 | Concat | 320 | 160×160 | 0 | 3 | 64 | FS Blcok_1 | 64 | 160×160 | 23 968 | 14 | 320 | C2f | 128 | 160×160 | 140 032 | 4 | 64 | FS Blcok_2 | 128 | 80×80 | 94 800 | 15 | 128 | Conv | 128 | 80×80 | 147 712 | 5 | 128 | FS Blcok_1 | 128 | 80×80 | 179 744 | 16 | 128 | Concat | 384 | 80×80 | 0 | 6 | 128 | FS Blcok_2 | 256 | 40×40 | 371 872 | 17 | 384 | C2f | 256 | 80×80 | 493 056 | 7 | 256 | FS Blcok_1 | 256 | 40×40 | 353 824 | 18 | 256 | Conv | 256 | 40×40 | 590 336 | 8 | 256 | SPPF | 256 | 40×40 | 164 608 | 19 | 256 | Concat | 512 | 40×40 | 0 | 9 | 256 | CARAFE | 256 | 80×80 | 103 012 | 20 | 512 | C2f | 256 | 40×40 | 525 824 | 10 | 256 | Concat | 384 | 80×80 | 0 | 21 | — | Detect | 3 | — | 1 677 550 | 11 | 384 | C2f | 256 | 80×80 | 493 056 | | | 总计 | | | 5 488 726 |
|
|
|