基于医学图像分类算法研究(三)——基于kvasir v2胃病数据集对transformer模型(vit,st,deit)以base基准型进行测试

前言:期间不断的在找更大规模更好的数据集,但是医学图像分类的数据集不好找,毕竟涉及隐私,目前分类的数据集没有比这个好的。要么是有很多张,但只有2类,一类有病一类没病。要么就是图片比较少。有些数据集还是其他格式的,不会用,哎。下面就展示下这个数据集对transformer模型以base基准型的训练效果吧。

本次测试换成6:2:2,感觉测试集太少了效果展现不佳,稍做了这个调整

下图是vit_base_patch16_224的测试,详细情况如下图

下图是swin_base_patch4_window7_224的测试,详细情况如下图

下图是deit_base_patch16_224的测试,详细情况如下图

根据上述情况对比,这3个模型针对该数据集效果最好的是swin-transformer的基准型

当然,各个模型其实还有很多分模型,由于设备有限只能测试3个给大家演示,后续看情况再做打算吧。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值