| # Data Module |
|
|
| ## Related codesοΌ |
|
|
| ``` |
| core/data/augments.py |
| core/data/dataloader.py |
| core/data/dataset.py |
| ``` |
|
|
| ## Dataset file format |
|
|
| In `LibContinual`, the dataset used has a fixed format. We read the data according to the dataset format set by most continual learning settings, such as [CIFAR-10](https://pytorch.org/vision/stable/datasets.html) and [CIFAR-100](https://pytorch.org/vision/stable/datasets.html). So we only need to download the dataset from the network and decompress it to use. If you want to use a new dataset and its data format is different from the above datasets, you need to convert it to the same format yourself. |
|
|
| Like CIFAR-10, the file format of the dataset should be the same as the following example: |
|
|
| ``` |
| dataset_folder/ |
| βββ train/ |
| βΒ Β βββ class_1/ |
| βΒ Β Β Β βββ image_1.png |
| β βββ ... |
| βΒ Β Β Β βββ image_5000.png |
| β βββ ... |
| βΒ Β βββ class_10/ |
| βΒ Β Β Β βββ image_1.png |
| β βββ ... |
| βΒ Β Β Β βββ image_5000.png |
| βββ test/ |
| βΒ Β βββ class_1/ |
| βΒ Β Β Β βββ image_1.png |
| β βββ ... |
| βΒ Β Β Β βββ image_5000.png |
| β βββ ... |
| βΒ Β βββ class_10/ |
| βΒ Β Β Β βββ image_1.png |
| β βββ ... |
| βΒ Β Β Β βββ image_5000.png |
| ``` |
|
|
| The training images and test images need to be placed in the `train` and `test` folders respectively, where all images of the same category are placed in folde with the same name as the category, such as `cat` , `dog`, etc. |
|
|
| ## Configure Datasets |
|
|
| After downloading or organizing the dataset according to the above file format, simply modify the `data_root` field in the configuration file. Note that `LibeContinual` will print the dataset folder name as the dataset name on the log. |
|
|