修改测试

This commit is contained in:
lee
2025-07-01 13:39:34 +08:00
parent 96a9445761
commit ed6a4144fa
8 changed files with 55 additions and 14 deletions

View File

@ -43,7 +43,7 @@ def get_image_list(path):
def _init():
model = YOLOv10('ckpts/20250620/best_gift_v10n.pt')
model = YOLOv10('ckpts/20250701/best_gift_v10n.pt')
return model
@ -94,8 +94,5 @@ def main(path):
if __name__ == "__main__":
# path = '../data_center/gift/trace_subimgs/d50' # 间距为50时
# path = '../data_center/gift/trace_subimgs/actual_test' # 永辉超市实测
path = '../data_center/gift/gift_test' #yolov10单图测试
# path = '../data_center/gift/trace_subimgs/tracluster' # tracluster方法过滤
main(path)