Skip to content

Commit

Permalink
README update
Browse files Browse the repository at this point in the history
  • Loading branch information
cobanov committed Jan 13, 2023
1 parent 9d1ac6b commit 182f3ce
Showing 1 changed file with 23 additions and 2 deletions.
25 changes: 23 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,5 1,26 @@
# Image Captioning

captioning is a Python library for dealing with image captioning using BLIP.

model_url = "https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_large_caption.pth"
model_url_vq = "https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model*_vqa.pth"
## Checkpoints [Required]

Download the fine-tuned checkpoint and copy into 'checkpoints' folder (create if does not exists)

- [BLIP-Large](https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_large_caption.pth)

## Usage

```
python inference.py --image_folder --output_folder
```

## Contributing

Pull requests are welcome. For major changes, please open an issue first
to discuss what you would like to change.

Please make sure to update tests as appropriate.

## License

[MIT](https://choosealicense.com/licenses/mit/)

0 comments on commit 182f3ce

Please sign in to comment.