caption_generator icon indicating copy to clipboard operation
caption_generator copied to clipboard

A modular library built on top of Keras and TensorFlow to generate a caption in natural language for any input image.

Results 28 caption_generator issues
Sort by recently updated
recently updated
newest added

After training, I noticed I just generate weights as follows: weights-improvement-01.hdf5 weights-improvement-02.hdf5 weights-improvement-03.hdf5 weights-improvement-04.hdf5 Apart from mentioned above, I get no other hdf5 file generated, can anyone tell me what's...

HI, I'm following your code and try to train the network on MSCOCO Here is my code class Caption_Model: def __init__(self,char_to_int,int_to_char,vocab_size=26688,max_caption_len=20,folder_path=path,epochs=10,batch_size=64): self.img_model=Sequential() self.text_model=Sequential() self.model=Sequential() self.vocab_size=vocab_size self.max_caption_len=max_caption_len self.folder_path=folder_path self.data={} self.char_to_int=char_to_int self.int_to_char=int_to_char...

Which version of keras is used? I'm getting error saying "module" object is not callable.

Hi! When I ran the train_model to train, keras did not use gpu automatically. However, when I ran mnist codes, the keras used the gpu automatically. I cant tell why....

Hi, after running prepare_dataset.py I tried to ran train_dataset.py. But it gives this error message. ``File "train_model.py", line 25, in train_model(epochs=50) File "train_model.py", line 6, in train_model cg = caption_generator.CaptionGenerator()...

Hello, I was wondering when do you think is worth to use a pretrained word embeddings model. I am facing a one-to-many problem as well, where my "many" are text...

added a functionality to validate the model while training

To fix error: TypeError: _obtain_input_shape() got an unexpected keyword argument 'include_top'. Without this change, to use vgg16.py, keras version should be degraded to 2.0.0.