caption_generator
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.
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.
Error