lichen/detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bakeup

Browse Source
This commit is contained in:
王庆刚
2024-09-02 11:50:08 +08:00
parent 5109400a57
commit 0cc36ba920
34 changed files with 1401 additions and 275 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
track_reid.py
View File

@ -172,7 +172,10 @@ def run(
if is_url and is_file:
source = check_file(source) # download
save_dir = Path(project) / Path(source).stem
# spth = source.split('\\')[-2] + "_" + Path(source).stem
save_dir = Path(project) / Path(source.split('\\')[-2] + "_" + str(Path(source).stem))
# save_dir = Path(project) / Path(source).stem
if save_dir.exists():
print(Path(source).stem)
# return
@ -387,6 +390,8 @@ def run(
# Print time (inference-only)
LOGGER.info(f"{s}{'' if len(det) else '(no detections), '}{dt[1].dt * 1E3:.1f}ms")
if track_boxes.size == 0:
return
## ======================================================================== written by WQG
## track_boxes: Array, [x1, y1, x2, y2, track_id, score, cls, frame_index, box_id]
@ -397,7 +402,7 @@ def run(
filename = os.path.split(save_path_img)[-1]
'''======================== 1. save in './run/detect/' ===================='''
if source.find("front") >= 0:
if source.find("front") >= 0 or Path(source).stem.split('_')[0] == '1':
carttemp = cv2.imread("./tracking/shopcart/cart_tempt/board_ftmp_line.png")
else:
carttemp = cv2.imread("./tracking/shopcart/cart_tempt/edgeline.png")
@ -516,10 +521,11 @@ def main_loop(opt):
optdict = vars(opt)
# p = r"D:\datasets\ym\永辉测试数据_比对"
p = r"D:\datasets\ym\广告板遮挡测试\8"
# p = r"D:\datasets\ym\广告板遮挡测试\8"
# p = r"D:\datasets\ym\videos\标记视频"
# p = r"D:\datasets\ym\实验室测试"
# p = r"D:\datasets\ym\永辉双摄视频\新建文件夹"
p = r"\\192.168.1.28\share\测试_202406\0723\0723_2\20240723-112522_"
k = 0
if os.path.isdir(p):
@ -531,16 +537,16 @@ def main_loop(opt):
# r"D:\datasets\ym\广告板遮挡测试\8\2500441577966_20240508-175946_front_addGood_70f75407b7ae_155_17788571404.mp4"
# ]
files = [r"D:\datasets\ym\广告板遮挡测试\8\6907149227609_20240508-174733_back_returnGood_70f754088050_425_17327712807.mp4"]
# files = [r"\\192.168.1.28\share\测试_202406\0723\0723_2\20240723-095838_\1_seek_193.mp4"]
for file in files:
optdict["source"] = file
run(**optdict)
k += 1
if k == 1:
break
# k += 1
# if k == 10:
# break
elif os.path.isfile(p):
optdict["source"] = p
run(**vars(opt))