➡️ Try it out on any movie poster here
Pre-trained model is available here
-
Download and extract the following datasets in a new folder called datasets:
-
Run
create_dataset.ipynb
to create train.csv and valid.csv -
Run
train.ipynb
to train the model