download: | | download: | | ||||
import json | import json | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
from utils.general import download, Path | from utils.general import download, Path | ||||
# Download script/URL (optional) --------------------------------------------------------------------------------------- | # Download script/URL (optional) --------------------------------------------------------------------------------------- | ||||
download: | | download: | | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
from utils.general import Path, check_requirements, download, np, xyxy2xywhn | from utils.general import Path, check_requirements, download, np, xyxy2xywhn | ||||
# Download script/URL (optional) --------------------------------------------------------------------------------------- | # Download script/URL (optional) --------------------------------------------------------------------------------------- | ||||
download: | | download: | | ||||
import shutil | import shutil | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
from utils.general import np, pd, Path, download, xyxy2xywh | from utils.general import np, pd, Path, download, xyxy2xywh | ||||
download: | | download: | | ||||
import xml.etree.ElementTree as ET | import xml.etree.ElementTree as ET | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
from utils.general import download, Path | from utils.general import download, Path | ||||
def visdrone2yolo(dir): | def visdrone2yolo(dir): | ||||
from PIL import Image | from PIL import Image | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
def convert_box(size, box): | def convert_box(size, box): | ||||
# Convert VisDrone box to YOLO xywh box | # Convert VisDrone box to YOLO xywh box |
import numpy as np | import numpy as np | ||||
from PIL import Image | from PIL import Image | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
from utils.datasets import autosplit | from utils.datasets import autosplit | ||||
from utils.general import download, xyxy2xywhn | from utils.general import download, xyxy2xywhn |
from torch.cuda import amp | from torch.cuda import amp | ||||
from torch.nn.parallel import DistributedDataParallel as DDP | from torch.nn.parallel import DistributedDataParallel as DDP | ||||
from torch.optim import SGD, Adam, AdamW, lr_scheduler | from torch.optim import SGD, Adam, AdamW, lr_scheduler | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
FILE = Path(__file__).resolve() | FILE = Path(__file__).resolve() | ||||
ROOT = FILE.parents[0] # YOLOv5 root directory | ROOT = FILE.parents[0] # YOLOv5 root directory |
import numpy as np | import numpy as np | ||||
import torch | import torch | ||||
import yaml | import yaml | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
from utils.general import LOGGER, colorstr, emojis | from utils.general import LOGGER, colorstr, emojis | ||||
import yaml | import yaml | ||||
from PIL import ExifTags, Image, ImageOps | from PIL import ExifTags, Image, ImageOps | ||||
from torch.utils.data import DataLoader, Dataset, dataloader, distributed | from torch.utils.data import DataLoader, Dataset, dataloader, distributed | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
from utils.augmentations import Albumentations, augment_hsv, copy_paste, letterbox, mixup, random_perspective | from utils.augmentations import Albumentations, augment_hsv, copy_paste, letterbox, mixup, random_perspective | ||||
from utils.general import (DATASETS_DIR, LOGGER, NUM_THREADS, check_dataset, check_requirements, check_yaml, clean_str, | from utils.general import (DATASETS_DIR, LOGGER, NUM_THREADS, check_dataset, check_requirements, check_yaml, clean_str, |
from typing import Dict | from typing import Dict | ||||
import yaml | import yaml | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
FILE = Path(__file__).resolve() | FILE = Path(__file__).resolve() | ||||
ROOT = FILE.parents[3] # YOLOv5 root directory | ROOT = FILE.parents[3] # YOLOv5 root directory |
import numpy as np | import numpy as np | ||||
import torch | import torch | ||||
from tqdm import tqdm | |||||
from tqdm.auto import tqdm | |||||
FILE = Path(__file__).resolve() | FILE = Path(__file__).resolve() | ||||
ROOT = FILE.parents[0] # YOLOv5 root directory | ROOT = FILE.parents[0] # YOLOv5 root directory |