multisource-docs-fix
UltralyticsAssistant 5 months ago
parent 92f2e12e05
commit e8676593ca
  1. 11
      ultralytics/data/loaders.py

@ -1,5 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
import csv
import glob
import math
import os
@ -14,7 +15,6 @@ import numpy as np
import requests
import torch
from PIL import Image
import csv
from ultralytics.data.utils import FORMATS_HELP_MSG, IMG_FORMATS, VID_FORMATS
from ultralytics.utils import IS_COLAB, IS_KAGGLE, LOGGER, ops
@ -275,12 +275,15 @@ class LoadImagesAndVideos:
def __init__(self, path, batch=1, vid_stride=1):
"""Initialize the Dataloader and raise FileNotFoundError if file not found."""
parent = None
if isinstance(path, str) and Path(path).suffix in [".txt", ".csv"]: # *.txt or *.csv file with img/vid/dir on each line
if isinstance(path, str) and Path(path).suffix in [
".txt",
".csv",
]: # *.txt or *.csv file with img/vid/dir on each line
parent = Path(path).parent
if Path(path).suffix == ".csv":
with open(path, 'r') as file:
with open(path) as file:
reader = csv.reader(file)
path = [row[0] for row in reader] # .csv file list of sources
path = [row[0] for row in reader] # .csv file list of sources
else:
path = Path(path).read_text().splitlines() # .txt file list of sources
files = []

Loading…
Cancel
Save