├── datasets/ # Directory for storing datasets ├── models/ # Pretrained models ├── scripts/ # Scripts for training, evaluation, and visualization │ ├── train.py # Script for training the model │ ├── ...