Skip to content

Commit

Permalink
get image file extension for later use
Browse files Browse the repository at this point in the history
  • Loading branch information
DarnellGranberry committed Nov 12, 2024
1 parent 8ddf2a5 commit 3417f45
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion topaz/extract.py
Original file line number Diff line number Diff line change
Expand Up @@ -333,7 +333,7 @@ def extract_particles(paths:List[str], model:Union[torch.nn.Module, str], device
for path,score,coords in nms_iterator(stream, radius, threshold, pool=pool, dims=dims, verbose=verbose):
# get the name of the image w/o extension
basename = os.path.basename(path)
name = os.path.splitext(basename)[0]
name, ext = os.path.splitext(basename)
if verbose:
report(f'Extracted {len(score)} particles from {name}')
# scale the coordinates
Expand Down

0 comments on commit 3417f45

Please sign in to comment.