|
|
@@ -28,7 +28,7 @@ from tqdm import tqdm |
|
|
|
|
|
|
|
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, |
|
|
|
cv2, segments2boxes, xyn2xy, xywh2xyxy, xywhn2xyxy, xyxy2xywhn) |
|
|
|
cv2, is_colab, is_kaggle, segments2boxes, xyn2xy, xywh2xyxy, xywhn2xyxy, xyxy2xywhn) |
|
|
|
from utils.torch_utils import torch_distributed_zero_first |
|
|
|
|
|
|
|
# Parameters |
|
|
@@ -321,6 +321,9 @@ class LoadStreams: |
|
|
|
import pafy |
|
|
|
s = pafy.new(s).getbest(preftype="mp4").url # YouTube URL |
|
|
|
s = eval(s) if s.isnumeric() else s # i.e. s = '0' local webcam |
|
|
|
if s == 0: |
|
|
|
assert not is_colab(), '--source 0 webcam unsupported on Colab. Rerun command in a local environment.' |
|
|
|
assert not is_kaggle(), '--source 0 webcam unsupported on Kaggle. Rerun command in a local environment.' |
|
|
|
cap = cv2.VideoCapture(s) |
|
|
|
assert cap.isOpened(), f'{st}Failed to open {s}' |
|
|
|
w = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) |