labels[:, :, 0] = -1 for i, target in enumerate(targets): if target.shape[0] > self.max_boxes_num: target = target[:self.max_boxes_num, :] labels[i][:target.shape[0]] = target target = target[: self.max_boxes_num, :] labels[i][: target.shape[0]] = target else: labe...
@zhongxu-Sun@Al-Razi-KRI was reviewing this. Our intention is to catch corrupted images/labels before training ever starts as part of the label caching process, so I think we may want to add additional checks for this failure mode. The proper workflow is that dataloader checks flag the corr...
from utils.general import labels_to_class_weights, increment_path, labels_to_image_weights, init_seeds, \ fitness, strip_optimizer, get_latest_run, check_dataset, check_file, check_git_status, check_img_size, \ print_mutation, set_logging from utils.google_utils import attempt_download ...