code and test set for article "<b>Single-shot full-Stokes imaging through scattering media</b>"

This repository provides the source code, test datasets, trained model weights, and predicted results associated with Figures 4 and 5 in the main manuscript. These figures illustrate AI-assisted metasurface imaging through scattering media. Due to differences in image naming conventions between the two experiments, the <code>split_data.py</code> script is slightly modified for each figure. We recommend running the code on a cloud-based environment to avoid potential dependency and configuration issues.