Skip to content

Latest commit

 

History

History
47 lines (23 loc) · 1.49 KB

README.md

File metadata and controls

47 lines (23 loc) · 1.49 KB

ThumbGenie

Thumbgenie will allow you to create custom YouTube thumbnails to use on your video, taking the title and category as inputs.

You can see some examples below:

thumbgenie demo

How to use:

Manually

  1. First, clone the repository locally

    git clone https://github.com/DylanJTodd/ThumbGenie
    
  2. Change into the directory created

    cd ./ThumbGenie/
    
  3. Next, pip install requirements.txt (Note, need python version 3.10+ )

    pip install -r requirements.txt
    
  4. If you want to train, then run preprocessing.py, followed by training.py. Ensure to tune the parameters such as epochs, batch size, resolution, etc, to your liking

    python preprocessing.py
    python training.py
    
  5. Run thumbgenie.py to start generating custom images. Please note, this file does come with a pretrained model, however due to limited resources, it was not trained thoroughly. I would recommend anyone with the ability to do so, to train their own model.

    python thumbgenie.py
    

Through Google Colab

  1. Open this notebook

  2. Make a personal copy, by going to the top left, through File > Save a copy in Drive

  3. Follow the instructions on the notebook.