https://fairytalefashionheadband.blogspot.com/2022/08/how-to-customize-pytorch-data.html