pytorch_multitask
dataset link here
Folders and files:
-fgo_multiclass_labels.csv has the class labels for the images in the images folder
-images folder contains the main dataset
-test folder is actually just a set of images I ran the model on to sense check it... but there are duplicates in the dataset so meh