ea83d29de00a46ab008455247fbdfc53182d747b,aocr/util/data_gen.py,DataGen,gen,#DataGen#Any#,54

Before Change

                        img_data = Image.open(StringIO(img))
                        width, height = img_data.size
                        resized_width = math.floor(float(width) / height * self.IMAGE_HEIGHT)

                        b_idx = min(resized_width, self.bucket_max_width)

                        bucket_size = self.bucket_data[b_idx].append(img, resized_width, word, lex)
                        if bucket_size >= batch_size:
                            bucket = self.bucket_data[b_idx].flush_out(

After Change

                            if bucket is not None:
                                yield bucket
                            else:
                                assert False, "No valid bucket."
                except tf.errors.OutOfRangeError:
                    break

        self.clear()

Instances