Is tracking multiple video sources using YOLO-NAS available?

[beykun18]

The code block below is taken from multi_yolo_backend.py.

def postprocess(self, path, preds, im ,im0s, predictor):
    if 'yolo_nas' in self.model_name or 'yolox' in self.model_name:
        predictor.results[0] = Results(
            path=path,
            boxes=preds,
            orig_img=im0s[0],
            names=predictor.model.names
        )
    else:
        predictor.results = predictor.postprocess(preds, im, im0s)
    return predictor.results

As you can see, it only return the result for the first image/video source. Is tracking multiple sources doable for YOLO-NAS or I interpreted the code wrongly? Thanks!

[mikel-brostrom]

You are right. Single stream only at the moment

[beykun18]

Thank you for answering.