Skip to content

Commit a580bd6

Browse files
authored
refactor: update normalization scheme
1 parent 51f8d0e commit a580bd6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/neuron_proofreader/merge_proofreading/merge_inference.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,7 @@ def read_superchunk(self, nodes):
347347
center = (start + shape // 2).astype(int)
348348
superchunk = self.img_reader.read(center, shape)
349349
superchunk = np.minimum(superchunk, self.brightness_clip)
350-
superchunk = img_util.normalize(superchunk)
350+
#superchunk = img_util.normalize(superchunk)
351351
return superchunk, start.astype(int)
352352

353353
def is_near_leaf(self, node, threshold=20):
@@ -516,7 +516,7 @@ def _get_multimodal_batch(self, nodes, img, offset):
516516
point_clouds = np.empty((len(nodes), 3, 3600), dtype=np.float32)
517517
for i, (node, center) in enumerate(zip(nodes, patch_centers)):
518518
s = img_util.get_slices(center, self.patch_shape)
519-
patches[i, 0, ...] = img[s]
519+
patches[i, 0, ...] = np.normalize(img[s])
520520
patches[i, 1, ...] = label_mask[s]
521521

522522
subgraph = self.graph.get_rooted_subgraph(

0 commit comments

Comments
 (0)