Browse Source

Use `tqdm.auto` (#7311)

modifyDataloader
Glenn Jocher GitHub 2 years ago
parent
commit
f735458987
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
11 changed files with 11 additions and 11 deletions
  1. +1
    -1
      data/Argoverse.yaml
  2. +1
    -1
      data/Objects365.yaml
  3. +1
    -1
      data/SKU-110K.yaml
  4. +1
    -1
      data/VOC.yaml
  5. +1
    -1
      data/VisDrone.yaml
  6. +1
    -1
      data/xView.yaml
  7. +1
    -1
      train.py
  8. +1
    -1
      utils/autoanchor.py
  9. +1
    -1
      utils/datasets.py
  10. +1
    -1
      utils/loggers/wandb/wandb_utils.py
  11. +1
    -1
      val.py

+ 1
- 1
data/Argoverse.yaml View File

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





+ 1
- 1
data/Objects365.yaml View File



# 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



+ 1
- 1
data/SKU-110K.yaml View File

# 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





+ 1
- 1
data/VOC.yaml View File

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





+ 1
- 1
data/VisDrone.yaml View File



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

+ 1
- 1
data/xView.yaml View File



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

+ 1
- 1
train.py View File

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

+ 1
- 1
utils/autoanchor.py View File

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



+ 1
- 1
utils/datasets.py View File

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,

+ 1
- 1
utils/loggers/wandb/wandb_utils.py View File

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

+ 1
- 1
val.py View File



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

Loading…
Cancel
Save