Error when next batch is to be loaded

costapt / vess2ret

A Keras implementation of pix2pix code adapted to generate retinal images from vessel networks.

MIT License

94 stars 42 forks source link

Error when next batch is to be loaded #13

Open divyagaddipati opened 6 years ago

divyagaddipati commented 6 years ago

indexarray, , current_batch_size = next(self.index_generator) ValueError: need more than 1 value to unpack

costapt commented 6 years ago

Hi!

What is the keras version you are using?

divyagaddipati commented 6 years ago

2.1.5

costapt commented 6 years ago

I am sorry, this code was developed with keras 1. If you wish to use keras 2 you will probably need to make some changes.

In keras 2, the index_generator only returns the index_array. I guess that you can easily get the current_batch_size by getting the lenght of the index_array!

divyagaddipati commented 6 years ago

Okay, changed it accordingly. Thanks.