Add global settings.yaml
in USER_CONFIG_DIR
(#125)
This commit is contained in:
@ -91,7 +91,7 @@ class BaseDataset(Dataset):
|
||||
# self.img_files = sorted([x for x in f if x.suffix[1:].lower() in IMG_FORMATS]) # pathlib
|
||||
assert im_files, f"{self.prefix}No images found"
|
||||
except Exception as e:
|
||||
raise Exception(f"{self.prefix}Error loading data from {img_path}: {e}\n{HELP_URL}")
|
||||
raise FileNotFoundError(f"{self.prefix}Error loading data from {img_path}: {e}\n{HELP_URL}") from e
|
||||
return im_files
|
||||
|
||||
def update_labels(self, include_class: Optional[list]):
|
||||
|
@ -484,7 +484,7 @@ class LoadImagesAndLabels(Dataset):
|
||||
# self.img_files = sorted([x for x in f if x.suffix[1:].lower() in IMG_FORMATS]) # pathlib
|
||||
assert self.im_files, f'{prefix}No images found'
|
||||
except Exception as e:
|
||||
raise Exception(f'{prefix}Error loading data from {path}: {e}\n{HELP_URL}') from e
|
||||
raise FileNotFoundError(f'{prefix}Error loading data from {path}: {e}\n{HELP_URL}') from e
|
||||
|
||||
# Check cache
|
||||
self.label_files = img2label_paths(self.im_files) # labels
|
||||
|
@ -12,10 +12,10 @@ import numpy as np
|
||||
import torch
|
||||
from PIL import ExifTags, Image, ImageOps
|
||||
|
||||
from ultralytics.yolo.utils import LOGGER, ROOT, colorstr
|
||||
from ultralytics.yolo.utils import LOGGER, ROOT, colorstr, yaml_load
|
||||
from ultralytics.yolo.utils.checks import check_file, check_font, is_ascii
|
||||
from ultralytics.yolo.utils.downloads import download
|
||||
from ultralytics.yolo.utils.files import unzip_file, yaml_load
|
||||
from ultralytics.yolo.utils.files import unzip_file
|
||||
|
||||
from ..utils.ops import segments2boxes
|
||||
|
||||
|
Reference in New Issue
Block a user