Skip to main content

Prepare Training Dataset

Organizing your data correctly is key for training classification models. ANSTS supports classification dataset in simple folder structure:

  • Create a main dataset folder: A folder named data to hold all your dataset
  • Create class folders: Inside your data folder, create a separate folder for each class. The folder name should be the class name your model will learn (e.g. Spiderman, Superman, Wonderwoman)
  • Add relevant images (jpg, png, or bmp) into their correct class folder; recommended 640 x 640 -pixel image at minimum

Example

Example dataset can be downloaded at Roboflow