Added more fonts, bold and italics

arrange_train_test_images.py

@@ -22,7 +22,7 @@ for font in fonts:
     os.makedirs(font_train_dir, exist_ok=True)
     os.makedirs(font_test_dir, exist_ok=True)
 
-    font_files = [f for f in os.listdir(GEN_IMAGES_DIR) if f.startswith(font)]
+    font_files = [f for f in os.listdir(GEN_IMAGES_DIR) if f.startswith(font + "_")]
     random.shuffle(font_files)
 
     train_files = font_files[:int(0.8 * len(font_files))]

consts.py

@@ -1,7 +1,29 @@
 # number of images to generate per font
 IMAGES_PER_FONT = 50
 # allowlist of fonts to use
-FONT_ALLOWLIST = ["Arial", "Avenir", "Courier", "Helvetica", "Georgia", "Tahoma", "Times New Roman", "Verdana"]
+FONT_ALLOWLIST = ["Arial",
+                  "Arial Black",
+                  "Arial Bold Italic",
+                  "Arial Bold",
+                  "Avenir", 
+                  "Courier",
+                  "Helvetica", 
+                  "Georgia", 
+                  "Tahoma",
+                  "Tahoma Bold",
+                  "Times New Roman",
+                  "Times New Roman Bold",
+                  "Times New Roman Italic",
+                  "Times New Roman Bold Italic",
+                  "Trebuchet MS",
+                  "Trebuchet MS Bold",
+                  "Trebuchet MS Italic",
+                  "Trebuchet MS Bold Italic",
+                  "Verdana",
+                  "Verdana Bold",
+                  "Verdana Italic",
+                  "Verdana Bold Italic"
+                  ]
 # directory where to store the generated images
 GEN_IMAGES_DIR = './generated_images'
 # images organized into train and test directories

train_font_identifier.py

@@ -77,7 +77,6 @@ def validate(model, data_loader, criterion):
 
 print(image_datasets['train'].classes)
 
-
 # Training loop with progress bar for epochs
 num_epochs = 10  # Replace with the number of epochs you'd like to train for
 for epoch in range(num_epochs):