Anchor-x/fed-yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix pylance type warning

Browse Source
This commit is contained in:
TY1667
2025-11-04 13:18:43 +08:00
parent f658c6aca5
commit e999853c94
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
utils/dataset.py
View File

@@ -101,6 +101,7 @@ class Dataset(data.Dataset):
return len(self.filenames) return len(self.filenames)
def load_image(self, i): def load_image(self, i):
# FIXME: for png maybe have something different with jpg
image = cv2.imread(self.filenames[i]) image = cv2.imread(self.filenames[i])
if image is None: if image is None:
raise FileNotFoundError(f"Image Not Found {self.filenames[i]}") raise FileNotFoundError(f"Image Not Found {self.filenames[i]}")
@@ -165,6 +166,13 @@ class Dataset(data.Dataset):
x2b = min(shape[1], x2a - x1a) x2b = min(shape[1], x2a - x1a)
y2b = min(y2a - y1a, shape[0]) y2b = min(y2a - y1a, shape[0])
if (
isinstance(x1a, type(None))
or isinstance(y1a, type(None))
or isinstance(x1b, type(None))
or isinstance(y1b, type(None))
):
raise ValueError("Mosaic calculation error")
pad_w = x1a - x1b pad_w = x1a - x1b
pad_h = y1a - y1b pad_h = y1a - y1b
image4[y1a:y2a, x1a:x2a] = image[y1b:y2b, x1b:x2b] image4[y1a:y2a, x1a:x2a] = image[y1b:y2b, x1b:x2b]
@@ -205,7 +213,7 @@ class Dataset(data.Dataset):
path = f"{os.path.dirname(filenames[0])}.cache" path = f"{os.path.dirname(filenames[0])}.cache"
if os.path.exists(path): if os.path.exists(path):
# XXX: temporarily disable cache # XXX: temporarily disable cache
os.remove(path) # os.remove(path)
pass pass
# return torch.load(path, weights_only=False) # return torch.load(path, weights_only=False)
x = {} x = {}
@@ -217,7 +225,10 @@ class Dataset(data.Dataset):
image.verify() # PIL verify image.verify() # PIL verify
shape = image.size # image size shape = image.size # image size
assert (shape[0] > 9) & (shape[1] > 9), f"image size {shape} <10 pixels" assert (shape[0] > 9) & (shape[1] > 9), f"image size {shape} <10 pixels"
if image.format:
assert image.format.lower() in FORMATS, f"invalid image format {image.format}" assert image.format.lower() in FORMATS, f"invalid image format {image.format}"
else:
assert any(filename.lower().endswith(f".{x}") for x in FORMATS), "unknown image format"
# verify labels # verify labels
a = f"{os.sep}images{os.sep}" a = f"{os.sep}images{os.sep}"