LOADING
513 words
3 minutes
路由多分支 YOLA 实验记录

指标#

FRBNet#

FRBNet 指标对比

ParadigmMethodExDark(YOLOv3)RecallExDark(YOLOv3)mAPExDark(TOOD)RecallExDark(TOOD)mAPDarkFace(YOLOv3)RecallDarkFace(YOLOv3)mAPDarkFace(TOOD)RecallDarkFace(TOOD)mAP
Baseline84.671.091.972.573.854.880.957.0
EnhancementSMG[66] (CVPR-23)82.368.591.871.573.452.480.256.3
NeRCo[70] (ICCV-23)83.468.591.871.873.853.079.456.8
LightDiff[25] (ECCV-24)84.371.392.172.975.557.481.058.7
DarkIR[15] (CVPR-25)81.968.290.972.074.555.981.460.4
Synthetic DataDAINet*[13] (CVPR-24)86.773.4--74.856.9--
WARLearn[1] (WACV-25)85.672.492.873.474.556.280.859.4
Multi-taskMAET[10] (ICCV-21)85.172.592.574.374.755.780.759.6
IAT[9] (BMVC-22)85.072.692.973.073.655.579.758.3
Plug-and-playDENet[46] (ACCV-22)84.271.392.673.571.852.673.649.6
FeatEnHancer[21] (ICCV-23)90.471.296.474.674.155.281.760.5
YOLA[5] (NeurIPS-24)86.172.793.875.274.956.383.163.2
FRBNet(NeurIPS-25)90.674.993.275.475.757.782.765.1

上图是FRBNet中的指标

下面我跑出来的测试指标,比论文图降低了0.2个百分点

yolov3-只开源了yolov3代码,tood怕补充错,不补充#

classgtsdetsrecallap
Bicycle41813530.9020.834
Boat51516070.8740.756
Bottle43320210.8500.750
Bus1645720.9510.886
Car91928450.9220.827
Cat42511500.8450.695
Chair60936780.8570.711
Cup35616970.8600.724
Dog49014980.9310.832
Motorbike24213510.8220.669
People223573910.8940.804
Table31129260.8230.483
mAP0.747

YOLA#

YOLA 指标对比

Exdark-iou_thr<0>.5#

TOOD#

类别GT数量检测数量RecallAP
Bicycle41835840.9330.833
Boat51554380.9400.785
Bottle43357930.8910.742
Bus16420420.9570.888
Car91990110.9330.794
Cat42537330.9080.749
Chair609101000.8930.706
Cup35652690.8990.722
Dog49041760.9630.862
Motorbike24253740.9050.633
People2235201780.9050.778
Table31189440.8840.495
mRecall0.918
mAP0.749

yolov3#

类别GT数量检测数量RecallAP
Bicycle41811590.8950.832
Boat51512750.8600.770
Bottle43314040.8180.726
Bus1644460.9390.881
Car91923270.8900.816
Cat42514070.8450.685
Chair60926480.7880.668
Cup35612010.8200.699
Dog49013450.9060.801
Motorbike24211820.7980.638
People223563570.8680.778
Table31121230.8070.494
mRecall0.853
mAP0.732

降点(得靠大batchsize来增点,点数也没之前高,离谱,之前不知道改啥了)

baseline#

classgtsdetsrecallap
Bicycle41813110.8660.801
Boat51512210.8170.722
Bottle43313820.8130.724
Bus1645110.9090.848
Car91924290.8620.783
Cat42513480.8310.669
Chair60929700.7720.651
Cup35610840.8310.711
Dog49012960.8980.786
Motorbike24214350.8260.641
People223561520.8580.768
Table31120790.7460.494
mAP0.717

DarkFace-iou_thr<0>.5#

TOOD(逆天成绩,之前没跑出来过,莫名其妙)#

类别GT数量检测数量RecallAP
Face4205493040.8680.711
mRecall0.868
mAP0.711

yolov3#

classgtsdetsrecallap
Face4205221150.7880.605
mAP0.605

OURS#

Exdark-iou_thr<0>.5#

TOOD#

类别GT数量检测数量RecallAP
Bicycle41833970.9330.843
Boat51557050.9320.797
Bottle43352470.8800.739
Bus16420940.9570.887
Car91982030.9380.801
Cat42537430.8820.731
Chair609106320.8920.698
Cup35650090.9070.708
Dog49039530.9730.880
Motorbike24249650.9050.645
People2235215880.9120.779
Table31187230.9040.509
mRecall0.918
mAP0.751

yolov3#

类别GT数量检测数量RecallAP
Bicycle41811600.8920.829
Boat51512640.8350.752
Bottle43314830.8040.706
Bus1644590.9330.878
Car91921690.8840.799
Cat42513670.8420.678
Chair60927450.7820.660
Cup35611750.8400.703
Dog49014030.9020.794
Motorbike24211230.8350.672
People223561840.8720.783
Table31121470.7750.492
mRecall0.850
mAP0.729

DarkFace-iou_thr<0>.5#

TOOD#

类别GT数量检测数量RecallAP
Face4205491200.8700.715
mRecall0.870
mAP0.715

yolov3#

类别GT数量检测数量RecallAP
Face4205218600.7960.622
mRecall0.796
mAP0.622

Ablation#

TOOD#

yola_hfe_exdark_ablation1_reflected_only#
类别GT数量检测数量RecallAP
Bicycle41833300.9350.837
Boat51550600.9360.788
Bottle43351270.8800.732
Bus16421790.9760.892
Car91991080.9360.795
Cat42536130.9040.731
Chair60999650.8830.700
Cup35650640.9020.705
Dog49041430.9630.871
Motorbike24252370.9260.660
People2235205090.9080.782
Table31193620.8970.506
mRecall0.920
mAP0.750
work_dirs/yola_hfe_exdark_ablation2_plus_frequency#
类别GT数量检测数量RecallAP
Bicycle41834080.9330.833
Boat51559270.9400.784
Bottle43358770.8710.729
Bus16420230.9630.879
Car91986690.9360.796
Cat42540030.8920.748
Chair609108840.9010.708
Cup35652100.9100.718
Dog49042190.9670.876
Motorbike24249700.9050.627
People2235214890.9080.778
Table31185640.8970.511
mRecall0.919
mAP0.749=
yola_hfe_exdark_ablation3_plus_wavelet#
类别GT数量检测数量RecallAP
Bicycle41834750.9400.835
Boat51560920.9380.785
Bottle43354300.8870.738
Bus16421000.9510.887
Car91987870.9300.793
Cat42538920.8960.747
Chair609104530.9010.706
Cup35652250.9040.715
Dog49040980.9690.880
Motorbike24249780.9210.644
People2235209630.9060.778
Table31185720.9000.512
mRecall0.920
mAP0.752
yola_hfe_exdark_ablation4_three_branches_no_attn#
类别GT数量检测数量RecallAP
Bicycle41832550.9280.831
Boat51559580.9300.784
Bottle43359810.8820.749
Bus16421690.9700.890
Car91988490.9370.805
Cat42537970.9060.747
Chair609101420.9010.699
Cup35652540.9040.731
Dog49041360.9710.876
Motorbike24251340.9210.634
People2235218250.9140.782
Table31185850.9000.492
mRecall0.922
mAP0.752
yola_hfe_exdark_ablation5_reflected_with_attn#
类别GT数量检测数量RecallAP
Bicycle41833390.9310.837
Boat51558550.9420.779
Bottle43353890.8730.741
Bus16421100.9510.892
Car91990190.9410.793
Cat42537470.9040.759
Chair609102700.8950.706
Cup35651780.9020.714
Dog49039610.9650.872
Motorbike24248550.9260.646
People2235204430.9100.779
Table31182960.8970.501
mRecall0.920
mAP0.752

新思路#

路由融合#

只使用yola原始的卷积融合融合多分支特征不能很好的根据样本做出分支特征的取舍,因此加入MoE的路由机制

使用mlp路由#

把两个分支的特征分别池化降采样然后展平成48维输入给mlp,输出为二维向量,分别对应分支权重,下面是yolov3的表现,和yola比降低了0.1个百分点

类别GT数量检测数量RecallAP
Bicycle41811230.8800.824
Boat51512340.8410.746
Bottle43314990.8270.734
Bus1644430.9330.881
Car91923690.8850.813
Cat42511850.8280.681
Chair60925850.7960.686
Cup35612170.8400.714
Dog49013060.9140.809
Motorbike24211040.7930.637
People223562740.8670.782
Table31121260.7720.467
mRecall0.848
mAP0.731

卷积做空间上的路由#

上面感觉B24608608B*24*608*608B48B*48作为路由输入信息损失太多,想着用卷积做Bbranchnums608608B*branchnums*608*608的路由,而且感觉只利用分支的特征信息做路由感觉不够有依据,按道理还要看原材料是怎么样的然后决定选择什么特征,因此就融入了RGB:B24608608RGB:B*24*608*608的特征作为路由输入。

# 3x3 卷积生成空间权重图
# 输入: (B, 48/72, H, W) → 输出: (B, 2, H, W)
self.router_conv = nn.Conv2d(
actual_in_channels,
num_experts,
kernel_size=kernel_size,
padding=padding,
bias=True
)

还是降点了,不多

类别GT数量检测数量RecallAP
Bicycle41811350.8850.822
Boat51511960.8390.748
Bottle43316080.8010.709
Bus1644340.9270.876
Car91921490.8820.808
Cat42513000.8330.676
Chair60925890.7820.659
Cup35611940.8600.722
Dog49013570.9140.783
Motorbike24210700.8060.647
People223564820.8640.771
Table31120700.7460.474
mRecall0.845
mAP0.725
路由多分支 YOLA 实验记录
/blog/posts/科研笔记/multi-branch-routing/
Author
Zenfish
Published at
2026-01-27
License
CC BY-NC-SA 4.0

Some information may be outdated