Skip to content

Commit 5ca584f

Browse files
authored
增加MultiScaleDataset的分割符 (#2853)
* 增加MultiScaleDataset的分割符 * remove encoding
1 parent 6cd88df commit 5ca584f

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

ppcls/data/dataloader/multi_scale_dataset.py

+13-2
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,25 @@
3030

3131

3232
class MultiScaleDataset(Dataset):
33+
"""MultiScaleDataset
34+
35+
Args:
36+
image_root (str): image root
37+
cls_label_path (str): path to annotation file `train_list.txt` or `val_list.txt`
38+
transform_ops (list, optional): list of transform op(s). Defaults to None.
39+
delimiter (str, optional): delimiter. Defaults to None.
40+
"""
41+
3342
def __init__(
3443
self,
3544
image_root,
3645
cls_label_path,
37-
transform_ops=None, ):
46+
transform_ops=None,
47+
delimiter=None, ):
3848
self._img_root = image_root
3949
self._cls_path = cls_label_path
4050
self.transform_ops = transform_ops
51+
self.delimiter = delimiter if delimiter is not None else " "
4152
self.images = []
4253
self.labels = []
4354
self._load_anno()
@@ -54,7 +65,7 @@ def _load_anno(self, seed=None):
5465
if seed is not None:
5566
np.random.RandomState(seed).shuffle(lines)
5667
for l in lines:
57-
l = l.strip().split(" ")
68+
l = l.strip().split(self.delimiter)
5869
self.images.append(os.path.join(self._img_root, l[0]))
5970
self.labels.append(np.int64(l[1]))
6071
assert os.path.exists(self.images[-1])

0 commit comments

Comments
 (0)