mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2026-09-19 01:53:56 +08:00
修改LaTeXOCR的数据处理部分,将生成的数据集中的绝对路径改为相对路径 (#13702)
* test * dataprocess_abspath2relpath
This commit is contained in:
@@ -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,
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user