__init__.py

import numpy as np
import torch
def track(tracker,args,orig_img,inps,boxes,hm,cropped_boxes,im_name,scores):
    hm = hm.cpu().data.numpy()
    online_targets = tracker.update(orig_img,inps,boxes,hm,cropped_boxes,im_name,scores,_debug=False)
    new_boxes,new_scores,new_ids,new_hm,new_crop = [],[],[],[],[]
    for t in online_targets:
        tlbr = t.tlbr
        tid = t.track_id
        thm = t.pose
        tcrop = t.crop_box
        tscore = t.detscore
        new_boxes.append(tlbr)
        new_crop.append(tcrop)
        new_hm.append(thm)
        new_ids.append(tid)
        new_scores.append(tscore)

    new_hm = torch.Tensor(new_hm).to(args.device)
    return new_boxes,new_scores,new_ids,new_hm,new_crop