Spaces:
Sleeping
Sleeping
西牧慧
commited on
Commit
·
ee15a41
1
Parent(s):
801a214
update: parcellation
Browse files
src/utils/parcellation.py
CHANGED
|
@@ -91,7 +91,7 @@ def parcellation(voxel, pnet_c, pnet_s, pnet_a, device):
|
|
| 91 |
torch.cuda.empty_cache()
|
| 92 |
|
| 93 |
# Combine the results from all views
|
| 94 |
-
out_e = out_a + out_e
|
| 95 |
del out_a
|
| 96 |
|
| 97 |
# Get the final parcellated output by taking the argmax
|
|
|
|
| 91 |
torch.cuda.empty_cache()
|
| 92 |
|
| 93 |
# Combine the results from all views
|
| 94 |
+
out_e = out_a + out_e
|
| 95 |
del out_a
|
| 96 |
|
| 97 |
# Get the final parcellated output by taking the argmax
|