Skip to content
Snippets Groups Projects

Update labels.json parsing during training

Merged Manon Blanco requested to merge label-parsing into main
1 file
+ 1
5
Compare changes
  • Side-by-side
  • Inline
@@ -95,15 +95,11 @@ class OCRDataset(Dataset):
set_name = path_and_set["set_name"]
gt = gt_per_set[set_name]
for filename in natural_sort(gt):
if isinstance(gt[filename], dict) and "text" in gt[filename]:
label = gt[filename]["text"]
else:
label = gt[filename]
filepath = Path(filename)
samples.append(
{
"name": filepath.name,
"label": label,
"label": gt[filename],
"path": filepath.resolve(),
}
)
Loading