训练过程报错 about yolov HOT 8 CLOSED

    preds_video = {}
    **repp_res = []**
    for frames in ele:
        # frames
        if frames == []: continue
        tmp_imgs = []
        for img in frames:
            img = cv2.imread(os.path.join(exp.data_dir, img))
            height, width = img.shape[:2]
            ratio = min(predictor.test_size[0] / img.shape[0], predictor.test_size[1] / img.shape[1])
            img, _ = predictor.preproc(img, None, predictor.test_size)
            img = torch.from_numpy(img)
            tmp_imgs.append(img)
        imgs = torch.stack(tmp_imgs)    
        
        pred_res = predictor.inference(imgs)
        del imgs
        for pred, img_name in zip(pred_res, frames):
            point_idx = img_name.rfind('.')
            image_id = img_name[img_name.find('val'):point_idx]
            img_idx = img_name[img_name.rfind('/') + 1:point_idx]
            det_repp = predictor.to_repp_heavy(pred, ratio, [height, width], image_id)
            preds_video[img_idx] = det_repp
    **repp_res = [video_name, preds_video]
    pickle.dump((repp_res [0], repp_res [1]), file_writter)
file_writter.close()**