Skip to content
Snippets Groups Projects
Commit a9ff4820 authored by Low, Low Jian He (PG/R - Comp Sci & Elec Eng)'s avatar Low, Low Jian He (PG/R - Comp Sci & Elec Eng)
Browse files

changing sorting to Phoenix specific

parent 88122573
No related branches found
No related tags found
No related merge requests found
......@@ -98,7 +98,12 @@ def main(args, model, renderer, device):
image_folder = Path(args.img_folder)
# Get all demo images ends with .jpg or .png
img_paths = [img for end in args.file_type for img in image_folder.glob(end)]
img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removesuffix('.png').removeprefix('Frame')))
# #MeinDGS Specific
# img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removesuffix('.png').removeprefix('Frame')))
#For Phoenix
img_paths = sorted(img_paths,key = lambda x: int(os.path.basename(x).removeprefix('images')))
img_cv2 = cv2.imread(str(img_paths[0]))
if args.bbox == 'True':
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment