这里放的是训练的脚本。 clean对数据进行了清晰,去除深度图或者无法打开的图片干扰。 ECA,SENet,CBAM为特征增强提取模块。加在Transformer类模型backbone前增强特征提取的能力。KAN为网MLP认为可以替换。 train_5k_origin为不添加卷积,只在各参数测试的最优化结果。(现在是dinov2) mixture是CBAM+SVM+RF投票的方法,代码有点bug(其余部分是另外一位同学添加,我不太清楚)。 qwen是想调用多模态打标的阴招。