pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

Implement labels argument for create_dataset (and downstream functions)

Open JosuaRieder opened this issue 10 months ago • 0 comments

Implement support for providing the labels with a json file for the "Folder" dataset. Support for other file formats could also be added easily.

I train on pseudo-labels and having to realize the pseudo-labeling using the file system is very slow.

It was not entirely clear to me at which level I should load the actual file, i.e. when to convert the Union[Dict, str] to Dict. I decided on create_reader but it could theoretically also be done one level later, i.e. in find_images_and_targets.

JosuaRieder avatar Jan 20 '25 01:01 JosuaRieder