search

nanoMFG / unet-sem

image segmentation for SVD graphene SEM images
MIT License
5 stars 1 forks source link

bug in booleen mask assignment #1

Closed dadamsncsa closed 5 years ago

dadamsncsa commented 5 years ago 
 model = Model(input = inputs, output = conv10)
Traceback (most recent call last):
  File "main.py", line 32, in <module>
    run.train()
  File "/home/dadams2/unet-sem/model.py", line 112, in train
    max_crop = self.max_crop)
  File "/home/dadams2/unet-sem/utils.py", line 65, in generate_batch
    out_mask[mask > 0.5] = 1
IndexError: boolean index did not match indexed array along dimension 0; dimension is 256 but corresponding boolean dimension is 1279