Spaces:
Sleeping
Sleeping
🐛 [Fix] Dataloader log error, remove format str
Browse files
yolo/tools/data_loader.py
CHANGED
|
@@ -103,7 +103,7 @@ class YoloDataset(Dataset):
|
|
| 103 |
img_path = images_path / image_name
|
| 104 |
data.append((img_path, labels))
|
| 105 |
valid_inputs += 1
|
| 106 |
-
logger.info("Recorded {}/{} valid inputs"
|
| 107 |
return data
|
| 108 |
|
| 109 |
def load_valid_labels(self, label_path: str, seg_data_one_img: list) -> Union[Tensor, None]:
|
|
|
|
| 103 |
img_path = images_path / image_name
|
| 104 |
data.append((img_path, labels))
|
| 105 |
valid_inputs += 1
|
| 106 |
+
logger.info(f"Recorded {valid_inputs}/{len(images_list)} valid inputs")
|
| 107 |
return data
|
| 108 |
|
| 109 |
def load_valid_labels(self, label_path: str, seg_data_one_img: list) -> Union[Tensor, None]:
|