Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

style_extract #3

Open
ama6eus opened this issue Apr 11, 2024 · 2 comments
Open

style_extract #3

ama6eus opened this issue Apr 11, 2024 · 2 comments

Comments

@ama6eus
Copy link

ama6eus commented Apr 11, 2024

Hi,@njucckevin:
I have some problem in the style_extract module. I can not find the bert_style_classifier model,"save_every_head-flickr7k-val.pt" and "save_every_head-senticap-val.pt". Besides, I can not find the code of "Similar Scenes arouse Similar Emotions" about style phrase extractor on the internet. Could you provide these models for referrence? I'd appreciate your help!

@njucckevin
Copy link
Owner

Hi,

Sorry I forgot to upload these checkpoints before. Now you can download these checkpoints in here, and place the save_every_head-flickr7k-val.pt, save_every_head-senticap-val.pt and two pytorch_model.bin in flickr7k and senticap respectively.

This style extractor are from paper Similar scenes arouse similar emotions: Parallel data augmentation for stylized image captioning. They constructed a style classifier base on Bert, and use the attention score to extract the style phrase.

@ama6eus
Copy link
Author

ama6eus commented Apr 12, 2024

Hi,

Sorry I forgot to upload these checkpoints before. Now you can download these checkpoints in here, and place the save_every_head-flickr7k-val.pt, save_every_head-senticap-val.pt and two pytorch_model.bin in flickr7k and senticap respectively.

This style extractor are from paper Similar scenes arouse similar emotions: Parallel data augmentation for stylized image captioning. They constructed a style classifier base on Bert, and use the attention score to extract the style phrase.

Thanks for your reply!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants