This repo is the implementation of "Image captioning for artworks". It currently includes my approach for this task.
git clone https://github.com/yangcandoit/BiasChecker.gitopen the ArtworkImageCaptioning folder
cd ArtworkImageCaptioning
python main.py --cfg configs/swin_tiny_patch4_window7_224.yaml --data_path /pathtodataset --batch_size 16| parameters | meaning |
|---|---|
| mode | train or test |
| exp_name | the name of this experiment |
| batch_size | the batch size of the data |
| data_path | the path of the dataset |
| resume | whether resume the model |
| output | output path. include the during training results and the configurations |