This project is an implementation of the Illustration2Vec model, which is a deep neural network for tagging and predicting attributes of illustrations. Here are the steps to run the project: With ACGANS
- Poetry init / poetry install
cd src/create_data/
and runpython generate_data_images.py
cd illustrationtovec
and runget_models.sh
- run
python get_tags.py
Note: This step may take several hours, depending on the size of your dataset. - Now
cd ..
and runpython create_csv.py
Now that you have the data and the images preproc, you May now run the main file or you May wish to run the other possible options in train / test directly.