search

broadinstitute / keras-rcnn

Keras package for region-based convolutional neural networks (RCNNs)
Other
555 stars 222 forks source link

Index Error #168

Open hallochen opened 6 years ago

hallochen commented 6 years ago

when inspect the trianingdata，this code( ....=generator.next() ） have an error ： IndexError: index 1 is out of bounds for axis 0 with size 1.

jhung0 commented 6 years ago

what training data?

hallochen commented 6 years ago

this is the code ：

training, validation, test = keras_rcnn.datasets.malaria_phenotypes.load_data() classes = { "rbc": 1, "leu": 2, "ring": 3, "tro": 4, "sch": 5, "gam": 6} generator = keras_rcnn.preprocessing.ObjectDetectionGenerator() generator = generator.flow(training, classes) (target_bounding_boxes, target_image, targetscores, ), _ = generator.next()

i want to get a training data ，but find this index error . then i modificate the _object_detection.py, but find another problem,