manually add class labels for application

utils.py

@@ -8,14 +8,26 @@ import torchvision.transforms as T
 from torchvision.utils import make_grid
 
 
+# default image directory (training only)
 img_path = '/home/alan/Projects/gen_dnd_art/filtered_images/im128/*pkl'
 img_files = glob.glob(img_path)
 
-# determine class names
+# determine class names from image directory (training only)
+'''
 labels = np.array([i.split('/')[-1].split('_')[:3] for i in img_files])
 species = np.unique(labels[:, 0]).tolist()
 classes = np.unique(labels[:, 1]).tolist()
 genders = np.unique(labels[:, 2]).tolist()
+'''
+
+# hard code class labels (for application)
+species = ['dragonborn', 'dwarf', 'elf', 'gnome', 'halfling', 'human', 'orc', 'tiefling']
+classes = [
+    'barbarian', 'bard', 'cleric', 'druid', 'fighter', 'monk',
+    'paladin','ranger', 'rogue', 'sorcerer', 'warlock', 'wizard'
+]
+genders = ['', 'female', 'male']
+
 
 class ImSet(Dataset):
     def __init__(self, img_path=img_path):