Skip to content

Commit 859f1cc

Browse files
author
yixu.cui
committed
Circumvent the problem of is_symlink
1 parent dc57198 commit 859f1cc

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

utils/datasets.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -660,7 +660,8 @@ def cache_labels(self, path=Path("./labels.cache"), prefix=""):
660660
try:
661661
# verify images
662662
im = Image.open(im_file)
663-
im.verify() # PIL verify
663+
if Path(im_file).is_symlink():
664+
im.verify() # PIL verify
664665
shape = exif_size(im) # image size
665666
segments = [] # instance segments
666667
assert (shape[0] > 9) & (shape[1] > 9), f"image size {shape} <10 pixels"
@@ -934,7 +935,8 @@ def __init__(
934935
self.path = path
935936

936937
# PREFIX = "/data/wujiapeng/datasets/"
937-
PREFIX = "/perception/yixu.cui/datas/tracking/"
938+
# PREFIX = "/perception/yixu.cui/datas/tracking/"
939+
PREFIX = "/share-global/yixu.cui/datas/tracking/"
938940
path = Path(path)
939941
assert path.is_file(), "wrong format for VisDrone"
940942

0 commit comments

Comments
 (0)