We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e2acbe3 commit 5ea8479Copy full SHA for 5ea8479
connectomics/engine/trainer.py
@@ -235,6 +235,8 @@ def test(self):
235
st = (np.array(st) *
236
np.array([1]+output_scale)).astype(int).tolist()
237
out_block = output[idx]
238
+ # hack it to make it positive for better weighted average
239
+ out_block[out_block < 0] = 0
240
if result[st[0]].ndim - out_block.ndim == 1: # 2d model
241
out_block = out_block[:, np.newaxis, :]
242
0 commit comments