# HG changeset patch # User Nicolas Saunier # Date 1694793376 14400 # Node ID 855abc69fa99b929435fe4767e78b0176be54f54 # Parent dd969637381ec76d14ab51b3c7568dbd5313ba71 forgot to use detector and tracker filename diff -r dd969637381e -r 855abc69fa99 scripts/dltrack.py --- a/scripts/dltrack.py Thu Sep 14 16:18:36 2023 -0400 +++ b/scripts/dltrack.py Fri Sep 15 11:56:16 2023 -0400 @@ -61,7 +61,7 @@ # check if one can go to specific frame https://docs.ultralytics.com/modes/track/#persisting-tracks-loop # Load a model -model = YOLO('/home/nicolas/Research/Data/classification-models/yolov8x.pt', ) # seg yolov8x-seg.pt +model = YOLO(args.detectorFilename, ) # seg yolov8x-seg.pt # seg could be used on cropped image... if can be loaded and kept in memory # model = YOLO('/home/nicolas/Research/Data/classification-models/yolo_nas_l.pt ') # AttributeError: 'YoloNAS_L' object has no attribute 'get' @@ -82,7 +82,7 @@ lastFrameNum = args.lastFrameNum success, frame = capture.read() -results = model.track(frame, tracker="/home/nicolas/Research/Data/classification-models/bytetrack.yaml", classes=list(moving.cocoTypeNames.keys()), persist=True) +results = model.track(frame, tracker=args.trackerFilename, classes=list(moving.cocoTypeNames.keys()), persist=True) # create object with user type and list of 3 features (bottom ones and middle) + projection while capture.isOpened() and success and frameNum <= lastFrameNum: #for frameNum, result in enumerate(results):