feat: initial HSAP platform
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
25
algorithms/lane_ufld/code/UFLD/get_picname.py
Executable file
25
algorithms/lane_ufld/code/UFLD/get_picname.py
Executable file
@@ -0,0 +1,25 @@
|
||||
import os
|
||||
import re
|
||||
|
||||
import cv2
|
||||
|
||||
rootdir = 'C:\\data\\Tusimple\\test_set\\51'
|
||||
pattern_number_oeder = '(\d*?).jpg'
|
||||
f1 = open('C:\\data\\Tusimple\\test_set\\51.txt', 'w')
|
||||
for root, dirs, files in os.walk(rootdir):
|
||||
for dir in dirs:
|
||||
print(os.path.join(root, dir))
|
||||
files.sort(key=lambda x:int(re.findall(pattern_number_oeder, x)[0]))
|
||||
for file in files:
|
||||
img_path = os.path.join(root, file)
|
||||
# img = cv2.imread(img_path)
|
||||
# dim = (1280, 720)
|
||||
# resized_img = cv2.resize(img, dim, interpolation=cv2.INTER_AREA)
|
||||
# cv2.imwrite(img_path, img)
|
||||
# save_path = os.path.join(root.split('\\')[-1], file)
|
||||
save_path = img_path
|
||||
print(save_path)
|
||||
# print(os.path.join(root, file))
|
||||
f1.write(save_path)
|
||||
f1.write('\n')
|
||||
f1.close()
|
||||
Reference in New Issue
Block a user