修改LaTeXOCR的数据处理部分,将生成的数据集中的绝对路径改为相对路径 (#13702)

* test

* dataprocess_abspath2relpath
This commit is contained in:
liuhongen1234567
2024-08-20 15:45:57 +08:00
committed by GitHub
parent 01e60ff9e1
commit 1752c56cb7
5 changed files with 8 additions and 3 deletions
+3 -1
View File
@@ -42,6 +42,7 @@ class LaTeXOCRDataSet(Dataset):
loader_config = config[mode]["loader"]
pkl_path = dataset_config.pop("data")
self.data_dir = dataset_config["data_dir"]
self.min_dimensions = dataset_config.pop("min_dimensions")
self.max_dimensions = dataset_config.pop("max_dimensions")
self.batchsize = dataset_config.pop("batch_size_per_pair")
@@ -128,7 +129,8 @@ class LaTeXOCRDataSet(Dataset):
images_transform = []
for img_path in ims:
for file_name in ims:
img_path = os.path.join(self.data_dir, file_name)
data = {
"img_path": img_path,
}