search

google / ffn

Flood-Filling Networks for instance segmentation in 3d volumes.
Apache License 2.0
318 stars 101 forks source link

ffn_inference_colab_demo issue #53

Closed ShixuanGu closed 1 year ago

ShixuanGu commented 1 year ago

IndexError Traceback (most recent call last) in <cell line: 6>() 4 else: 5 vis_update = 1 ----> 6 canvas.segment_at((125, 125, 125), dynamic_image=inference.DynamicImage(),vis_update_every=vis_update)

1 frames /content/ffn/ffn/inference/inference.py in update_at(self, pos, start_pos) 411 end = start + self._input_seed_size 412 logit_seed = np.array( --> 413 self.seed[[slice(s, e) for s, e in zip(start, end)]]) 414 init_prediction = np.isnan(logit_seed) 415 logit_seed[init_prediction] = np.float32(self.options.pad_value)

IndexError: only integers, slices (:), ellipsis (...), numpy.newaxis (None) and integer or boolean arrays are valid indices

mjanusz commented 1 year ago

Thank you for the bug report. This has been fixed now -- sorry about the breakage!