main_mae_train.py 这个文件是为了训练补全模型,主要的目的是将图片中猪被遮挡的部分进行还原。
main_mae_weight.py 这个文件是将补全模型和称重模型放到一起进行训练,主要的目的是将图片中猪被遮挡的部分进行还原,并且将猪的重量进行预测。
main_weight_train.py 这个文件是为了训练称重模型,主要的目的是将猪的重量进行预测。
- 补全模型的训练,完成
- 称重模型的训练,完成
- 补全+称重模型的训练,完成
- 称重模型在V3测试集上的测试,也就是称重模型在mask数据集上的测试,看看mask对称重模型的影响
- 补全模型在V3测试集上的推理,也就是将测试集合中的mask进行补全,然后将补全后的图片保存下来,然后使用称重模型进行预测,看看补全后的图像对称重模型的影响,完成