Sylwia Mielnicka

Visualisation, AI and data analysis



PyTorch labels form CSV – custom DataLoader

In PyTorch the labels are generated by default form subfolder names in the root directory. It’s very comfortable unless you’d like to change the labels. Let’s say, I ‘d like to train the two-staged Image Recognition model: Predict if an… Continue Reading →

© 2020 Sylwia Mielnicka — Powered by WordPress

Theme by Anders NorenUp ↑