sugi-chan / pytorch_multitask

multitask model resnet 50

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

multitask model resnet 50


Languages

Language:Jupyter Notebook 100.0%