[飞桨黑客松第四期] (127) MobileViTv2

[flytocc]

任务详情： [飞桨黑客松第四期任务总览] Paddle#51281

修改内容

1. [bugfix] 当DecodeImage使用pil backend时，image类型应转为RGB
2. [bugfix] 在MixupOperator、CutmixOperator和FmixOperator中，索引打乱的方法存在一定缺陷，np.random.permutation可能造成部分索引在打乱前后顺序不变。改为逆序打乱索引。
3. [bugfix] 当使用checkpoint恢复训练时，更新sampler中的epoch为当前的epoch
4. [update] 为RandCropImage添加use_log_aspect的支持
5. [add] 增加RandAugmentV3，用于MobileViTv2模型的训练
6. [add] 增加MobileViTv2算法，包括model、config、test_tipc、doc

复现精度

测试

Models	Top1	Top5	Reference top1	Reference top5	FLOPs (G)	Params (M)
MobileViTv2_x0_5	0.7017	0.8988	0.7018	-	480.46	1.37
MobileViTv2_x1_0	0.7813	0.9417	0.7809	-	1843.81	4.90
MobileViTv2_x1_5	0.8034	0.9509	0.8038	-	4090.07	10.60
MobileViTv2_x2_0	0.8116	0.9537	0.8117	-	7219.23	18.45

训练

训练平台为AIStudio，PaddlePaddle2.4.0 with V100*4

Models	Top1	Top5	Reference top1	Reference top5	weight & log
MobileViTv2_x1_5	0.8045	0.9503	0.8038	-	best_model_ema.ema.params & train*.log

权重及训练日志下载地址：百度网盘

[Ligoml]

@weisy11 请review下这道黑客松题目 谢谢

[TingquanGao]

验收通过：
MobileViT_x1_5: top1 acc 0.8038

[TingquanGao]

LGTM