images/: test images
model/: pretrained models, including image feature extraction model (object.params, scene.params, Sentiment.params) and poem generation model (ckpt/)
src/: code for testing
- test.py
To test how much time it cost to generate a poem for an image
def get_poem(image_file):
"""Generate a poem from the image whose filename is `image_file`
Parameters
----------
image_file : str
Path to the input image
Returns
-------
str
Generated Poem
"""