Skip to content

Commit 7c1f595

Browse files
committed
Fix SGN subtype input
1 parent e20b8e1 commit 7c1f595

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

scripts/intensity_annotation/sgn_subtype_annotation.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -137,11 +137,12 @@ def _create_mask(seg_extended, stain):
137137
def sgn_subtype_annotation(prefix, subtype, seg_version="SGN_v2", default_stat="median"):
138138

139139
direc = os.path.dirname(os.path.abspath(prefix))
140+
basename = os.path.basename(prefix)
140141
file_names = [entry.name for entry in os.scandir(direc)]
141142

142-
stain1_file = [name for name in file_names if prefix in name and subtype in name][0]
143-
stain2_file = [name for name in file_names if prefix in name and "PV.tif" in name][0]
144-
seg_file = [name for name in file_names if prefix in name and "SGN" in name][0]
143+
stain1_file = [name for name in file_names if basename in name and subtype in name][0]
144+
stain2_file = [name for name in file_names if basename in name and "PV.tif" in name][0]
145+
seg_file = [name for name in file_names if basename in name and "SGN" in name][0]
145146

146147
if "PV" in seg_file:
147148
seg_version = f"PV_{seg_version}"

0 commit comments

Comments
 (0)