Skip to content

Commit f41ff1d

Browse files
author
donglaiw
committed
update valid count
1 parent 3fb386a commit f41ff1d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

connectomics/data/dataset/dataset_volume.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,9 @@ def __init__(self,
143143
for i, x in enumerate(self.valid_mask):
144144
if x is not None:
145145
self.valid_pos[i] = get_valid_pos(x, sample_volume_size, valid_ratio)
146+
self.sample_num[i] = self.valid_pos[i].shape[0]
147+
self.sample_num_a = np.sum(self.sample_num)
148+
self.sample_num_c = np.cumsum([0] + list(self.sample_num))
146149

147150
if self.mode in ['val', 'test']: # for validation and test
148151
self.sample_size_test = [

0 commit comments

Comments
 (0)