feat: randomize generator

font_dataset/background.py

@@ -1,4 +1,5 @@
 import os
+import random
 from PIL import Image
 
 
@@ -10,5 +11,4 @@ def background_image_generator(path="./dataset/pixivimages"):
     image_list = [os.path.join(path, image) for image in image_list]
 
     while True:
-        for image in image_list:
-            yield image
+        yield random.choice(image_list)

font_dataset/text.py

@@ -77,11 +77,9 @@ class JapaneseUtaNetCorpusGenerator(CommonCorpusGenerator):
         self.conn = sqlite3.connect("lyrics_corpus/cache/uta-net.db")
         self.cur = self.conn.cursor()
 
-        i = 0
-
         while True:
             self.cur.execute(
-                f"SELECT lyrics FROM lyrics ORDER BY song_id LIMIT 1 OFFSET {i}"
+                "SELECT lyrics FROM lyrics WHERE song_id IN (SELECT song_id FROM lyrics ORDER BY RANDOM() LIMIT 1)"
             )
             row = self.cur.fetchone()
             if row is not None:
@@ -92,7 +90,6 @@ class JapaneseUtaNetCorpusGenerator(CommonCorpusGenerator):
                     continue
             else:
                 return
-            i += 1
 
     def _random_place_holder(self, font: DSFont) -> str:
         r = random.randint(1, 3)