https://camera-optical-zoom.blogspot.com/2024/06/how-to-customize-pytorch-data.html