news-navigator/code/code-merge-v6/cfg.yaml
2022-08-02 13:18:36 +00:00

10 lines
423 B
YAML

path: /mnt/gpu_data1/kubapok/news-navigator-annex/news-navigator # dataset root dir
train: train-dataset # train images (relative to 'path') 128 images
val: dev-0-dataset # val images (relative to 'path') 128 images
test: test-A-dataset # test images (optional)
# Classes
nc: 8 # number of classes
names: ['photograph', 'illustration', 'map', 'cartoon', 'editorial_cartoon', 'headline', 'advertisement', 'combined']