# -*- coding: utf-8 -*- from words_image import save_image import os base_dir = os.path.dirname(__file__) words_dir = os.path.join(base_dir, "data") image_name = os.path.join(base_dir, "tony.png") if __name__ == "__main__": files = [] for f in os.listdir(words_dir): full_f = os.path.join(words_dir, f) if os.path.isfile(full_f): files.append(full_f) save_image(files, image_name)
# -*- coding: utf-8 -*- from words_image import get_keywords, save_image import os BASE_DIR = os.path.dirname(__file__) DATA_DIR = os.path.join(BASE_DIR, "data") IMAGE_DIR = os.path.join(BASE_DIR, "images") OUTPUT_DIR = os.path.join(BASE_DIR, "output") if __name__ == "__main__": words = get_keywords(DATA_DIR) for image in os.listdir(IMAGE_DIR): image_name = os.path.join(IMAGE_DIR, image) save_image(words, image_name, OUTPUT_DIR)