一种基于神经网络的偏滤器可见光相机图像的快速反演方法及装置

本发明公开了一种基于神经网络的偏滤器可见光相机图像的快速反演方法及装置，属于图像断层重建技术领域，该方法包括：构建初步的数据集；根据初步数据集选取主要反映中性粒子及杂质辐射的区域，裁剪固定的大小作为真正的数据集；在假设托卡马克等离子体环向对称的情况下，计算权重矩阵；将真正数据集划分训练集、验证集和测试集；训练集放入ViT模型进行训练，ViT模型的输出为反演获得的中性粒子及杂质辐射分布；之后根据相机图像与中性粒子及杂质的辐射分布之间的关系，将中性粒子及杂质的辐射分布与的权重矩阵相乘得到模拟的相机图像，根据网络的总体损失训练ViT神经网络。本发明实现了一次计算得出辐射分布，节约了运算资源。