MCPcopy Create free account

hub / github.com/DataXujing/YOLOv8 / functions

Functions814 in github.com/DataXujing/YOLOv8

↓ 1 callersMethod_update_labels
Update labels
ultralytics/yolo/data/augment.py:158
↓ 1 callersMethod_update_labels
Update labels
ultralytics/yolo/data/augment.py:486
↓ 1 callersMethodaffine_transform
(self, img)
ultralytics/yolo/data/augment.py:213
↓ 1 callersMethodapply_bboxes
apply affine to bboxes only. Args: bboxes(ndarray): list of bboxes, xyxy format, with shape (num_bboxes, 4). M(ndarra
ultralytics/yolo/data/augment.py:253
↓ 1 callersMethodapply_image
(self, labels)
ultralytics/yolo/data/augment.py:26
↓ 1 callersMethodapply_instances
(self, labels)
ultralytics/yolo/data/augment.py:29
↓ 1 callersMethodapply_keypoints
apply affine to keypoints. Args: keypoints(ndarray): keypoints, [N, 17, 2]. M(ndarray): affine matrix. Return
ultralytics/yolo/data/augment.py:299
↓ 1 callersMethodapply_segments
apply affine to segments and generate new bboxes from segments. Args: segments(ndarray): list of segments, [num_samples, 500, 2].
ultralytics/yolo/data/augment.py:276
↓ 1 callersMethodapply_semantic
(self, labels)
ultralytics/yolo/data/augment.py:32
↓ 1 callersMethodareas
(self)
ultralytics/yolo/utils/instance.py:79
↓ 1 callersFunctionaugment_hsv
(im, hgain=0.5, sgain=0.5, vgain=0.5)
ultralytics/yolo/data/dataloaders/v5augmentations.py:69
↓ 1 callersMethodauth_with_cookies
Attempt to fetch authentication via cookies and set id_token. User must be logged in to HUB and running in a supported browser.
ultralytics/hub/auth.py:38
↓ 1 callersFunctionautobatch
(model, imgsz=640, fraction=0.7, batch_size=16)
ultralytics/yolo/utils/autobatch.py:21
↓ 1 callersFunctionbbox2dist
Transform bbox(xyxy) to dist(ltrb).
ultralytics/yolo/utils/tal.py:208
↓ 1 callersMethodbias_init
(self)
ultralytics/nn/modules.py:642
↓ 1 callersFunctionbox_candidates
(box1, box2, wh_thr=2, ar_thr=100, area_thr=0.1, eps=1e-16)
ultralytics/yolo/data/dataloaders/v5augmentations.py:302
↓ 1 callersMethodbox_candidates
(self, box1, box2, wh_thr=2, ar_thr=100, area_thr=0.1, eps=1e-16)
ultralytics/yolo/data/augment.py:370
↓ 1 callersMethodbuild_optimizer
> Builds an optimizer with the specified parameters and parameter groups. Args: model (nn.Module): model to optimize
ultralytics/yolo/engine/trainer.py:534
↓ 1 callersMethodbuild_transforms
Users can custom augmentations here like: if self.augment: # training transforms return Compose([]
ultralytics/yolo/data/base.py:200
↓ 1 callersMethodcache_images
(self, cache)
ultralytics/yolo/data/base.py:132
↓ 1 callersMethodcache_labels
(self, path=Path("./labels.cache"))
ultralytics/yolo/data/dataset.py:48
↓ 1 callersMethodcache_labels
(self, path=Path('./labels.cache'), prefix='')
ultralytics/yolo/data/dataloaders/v5loader.py:610
↓ 1 callersMethodcheck_cache_ram
(self, safety_margin=0.1, prefix='')
ultralytics/yolo/data/dataloaders/v5loader.py:593
↓ 1 callersFunctioncheck_dataset_disk_space
(url='https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip', sf=2.0)
ultralytics/hub/utils.py:17
↓ 1 callersMethodcheck_disk_space
(self)
ultralytics/hub/session.py:99
↓ 1 callersFunctioncheck_imshow
(warn=False)
ultralytics/yolo/utils/checks.py:232
↓ 1 callersFunctioncheck_online
Check internet connectivity by attempting to connect to a known online host. Returns: bool: True if connection is successful, False
ultralytics/yolo/utils/checks.py:127
↓ 1 callersFunctioncheck_pil_font
(font=FONT, size=10)
ultralytics/yolo/utils/plotting.py:141
↓ 1 callersFunctioncheck_python
Check current python version against the required minimum version. Args: minimum (str): Required minimum version of python. Ret
ultralytics/yolo/utils/checks.py:143
↓ 1 callersMethodcheck_resume
(self)
ultralytics/yolo/engine/trainer.py:498
↓ 1 callersMethodcheck_stats
(self, stats)
ultralytics/yolo/engine/validator.py:200
↓ 1 callersFunctioncheck_train_batch_size
(model, imgsz=640, amp=True)
ultralytics/yolo/utils/autobatch.py:15
↓ 1 callersFunctionclassify_albumentations
( augment=True, size=224, scale=(0.08, 1.0), hflip=0.5, vflip=0.0,
ultralytics/yolo/data/augment.py:691
↓ 1 callersFunctionclassify_albumentations
( augment=True, size=224, scale=(0.08, 1.0), ratio=(0.75, 1.0 / 0.75), # 0.75
ultralytics/yolo/data/dataloaders/v5augmentations.py:310
↓ 1 callersFunctionclassify_transforms
(size=224)
ultralytics/yolo/data/augment.py:684
↓ 1 callersFunctionclassify_transforms
(size=224)
ultralytics/yolo/data/dataloaders/v5augmentations.py:352
↓ 1 callersFunctionclean_str
(s)
ultralytics/yolo/utils/ops.py:672
↓ 1 callersFunctionclip_coords
(boxes, shape)
ultralytics/yolo/utils/ops.py:251
↓ 1 callersFunctionclip_segments
> It takes a list of line segments (x1,y1,x2,y2) and clips them to the image shape (height, width) Args: segments: a list of segments,
ultralytics/yolo/utils/ops.py:655
↓ 1 callersMethodclose_mosaic
(self, hyp)
ultralytics/yolo/data/dataset.py:141
↓ 1 callersFunctioncompute_ap
Compute the average precision, given the recall and precision curves # Arguments recall: The recall curve (list) precision: Th
ultralytics/yolo/utils/metrics.py:318
↓ 1 callersFunctioncreate_and_add_plugin_node
(graph, topK, keepTopK)
tensorrt/yolov8_add_nms.py:12
↓ 1 callersFunctioncreate_attrs
(topK, keepTopK)
tensorrt/yolov8_add_nms.py:42
↓ 1 callersFunctionddp_cleanup
(command, trainer)
ultralytics/yolo/utils/dist.py:58
↓ 1 callersMethodeval_json
(self, stats)
ultralytics/yolo/engine/validator.py:223
↓ 1 callersFunctionexif_size
(img)
ultralytics/yolo/data/utils.py:53
↓ 1 callersFunctionexif_size
(img)
ultralytics/yolo/data/dataloaders/v5loader.py:62
↓ 1 callersFunctionexport
(cfg)
ultralytics/yolo/engine/exporter.py:814
↓ 1 callersMethodfinal_eval
(self)
ultralytics/yolo/engine/trainer.py:487
↓ 1 callersFunctionfind_free_network_port
Finds a free port on localhost. It is useful in single-node training when we don't want to connect to a real main node but have to set the `M
ultralytics/yolo/utils/dist.py:12
↓ 1 callersMethodforward
Runs inference on the given model Args: im: the image tensor augment: whether to augment the image. Defaults to
ultralytics/nn/autobackend.py:234
↓ 1 callersMethodfuse
(self)
ultralytics/yolo/engine/model.py:110
↓ 1 callersFunctionfuse_conv_and_bn
(conv, bn)
ultralytics/yolo/utils/torch_utils.py:104
↓ 1 callersFunctiongenerate_ddp_command
(world_size, trainer)
ultralytics/yolo/utils/dist.py:47
↓ 1 callersFunctiongenerate_ddp_file
(trainer)
ultralytics/yolo/utils/dist.py:26
↓ 1 callersMethodget_annotator
(self, img)
ultralytics/yolo/v8/classify/predict.py:14
↓ 1 callersMethodget_box_metrics
(self, pd_scores, pd_bboxes, gt_labels, gt_bboxes)
ultralytics/yolo/utils/tal.py:124
↓ 1 callersMethodget_dataloader
(self, dataset_path, batch_size)
ultralytics/yolo/engine/validator.py:182
↓ 1 callersMethodget_dataset
> Get train, val path from data dict if it exists. Returns None if data format is not recognized.
ultralytics/yolo/engine/trainer.py:376
↓ 1 callersFunctionget_default_args
(func)
ultralytics/yolo/utils/__init__.py:191
↓ 1 callersMethodget_desc
(self)
ultralytics/yolo/engine/validator.py:206
↓ 1 callersFunctionget_git_root_dir
Determines whether the current file is part of a git repository and if so, returns the repository root directory. If the current file is not
ultralytics/yolo/utils/__init__.py:179
↓ 1 callersMethodget_img_files
Read image files.
ultralytics/yolo/data/base.py:75
↓ 1 callersMethodget_indexes
(self)
ultralytics/yolo/data/augment.py:99
↓ 1 callersMethodget_indexes
(self)
ultralytics/yolo/data/augment.py:189
↓ 1 callersMethodget_labels
Users can custom their own format here. Make sure your output is a list with each element like below: dict( im_fil
ultralytics/yolo/data/base.py:212
↓ 1 callersFunctionget_latest_run
(search_dir='.')
ultralytics/yolo/utils/files.py:100
↓ 1 callersMethodget_model
(self, cfg=None, weights=None, verbose=True)
ultralytics/yolo/v8/classify/train.py:26
↓ 1 callersFunctionget_num_gradients
(model)
ultralytics/yolo/utils/torch_utils.py:149
↓ 1 callersMethodget_pos_mask
(self, pd_scores, pd_bboxes, gt_labels, gt_bboxes, anc_points, mask_gt)
ultralytics/yolo/utils/tal.py:111
↓ 1 callersFunctionget_settings
Loads a global settings YAML file or creates one with default values if it does not exist. Args: file (Path): Path to the settings Y
ultralytics/yolo/utils/__init__.py:347
↓ 1 callersMethodget_state
Get the authentication state
ultralytics/hub/auth.py:64
↓ 1 callersMethodget_stats
(self)
ultralytics/yolo/engine/validator.py:197
↓ 1 callersMethodget_targets
Args: gt_labels: (b, max_num_obj, 1) gt_bboxes: (b, max_num_obj, 4) target_gt_idx: (b, h*w) f
ultralytics/yolo/utils/tal.py:155
↓ 1 callersFunctionget_user_config_dir
Get the user config directory. Args: sub_dir (str): The name of the subdirectory to create. Returns: Path: The path to
ultralytics/yolo/utils/__init__.py:197
↓ 1 callersMethodget_validator
(self)
ultralytics/yolo/engine/trainer.py:439
↓ 1 callersFunctiongit_describe
(path=ROOT)
ultralytics/yolo/utils/checks.py:248
↓ 1 callersMethodhex2rgb
(h)
ultralytics/yolo/utils/plotting.py:36
↓ 1 callersFunctionimg2label_paths
(img_paths)
ultralytics/yolo/data/utils.py:39
↓ 1 callersMethodinit_metrics
(self, model)
ultralytics/yolo/engine/validator.py:191
↓ 1 callersFunctioninit_seeds
(seed=0, deterministic=False)
ultralytics/yolo/utils/torch_utils.py:227
↓ 1 callersFunctioninitialize_weights
(model)
ultralytics/yolo/utils/torch_utils.py:167
↓ 1 callersFunctionis_dir_writeable
Check if a directory is writeable. Args: dir_path (str): The path to the directory. Returns: bool: True if the director
ultralytics/yolo/utils/__init__.py:161
↓ 1 callersFunctionis_docker
Determine if the script is running inside a Docker container. Returns: bool: True if the script is running inside a Docker container
ultralytics/yolo/utils/__init__.py:115
↓ 1 callersFunctionis_jupyter_notebook
Check if the current script is running inside a Jupyter Notebook. Verified on Colab, Jupyterlab, Kaggle, Paperspace. Returns: bo
ultralytics/yolo/utils/__init__.py:98
↓ 1 callersFunctionis_parallel
(model)
ultralytics/yolo/utils/torch_utils.py:212
↓ 1 callersFunctionis_url
(url, check=True)
ultralytics/yolo/utils/downloads.py:39
↓ 1 callersMethodload
(self, weights, verbose=True)
ultralytics/nn/tasks.py:213
↓ 1 callersMethodload
(self, weights)
ultralytics/nn/tasks.py:266
↓ 1 callersMethodload_image
(self, i)
ultralytics/yolo/data/base.py:115
↓ 1 callersFunctionltwh2xywh
> Convert nx4 boxes from [x1, y1, w, h] to [x, y, w, h] where xy1=top-left, xy=center Args: x: the input tensor
ultralytics/yolo/utils/ops.py:434
↓ 1 callersFunctionltwh2xyxy
> It converts the bounding box from [x1, y1, w, h] to [x1, y1, x2, y2] where xy1=top-left, xy2=bottom-right Args: x: the input ima
ultralytics/yolo/utils/ops.py:447
↓ 1 callersFunctionmain
()
tensorrt/yolov8_add_nms.py:76
↓ 1 callersFunctionmake_divisible
(x, divisor)
ultralytics/yolo/utils/ops.py:97
↓ 1 callersMethodmap
Mean AP@0.5:0.95 of all classes. Return: float.
ultralytics/yolo/utils/metrics.py:463
↓ 1 callersFunctionmask_iou
mask1: [N, n] m1 means number of predicted objects mask2: [M, n] m2 means number of gt objects Note: n means image_w x image_h return
ultralytics/yolo/utils/metrics.py:107
↓ 1 callersMethodmasks
Plot masks at once. Args: masks (tensor): predicted masks on cuda, shape: [n, h, w] colors (List[List[Int]]): colors f
ultralytics/yolo/utils/plotting.py:89
↓ 1 callersMethodmean_results
(self)
ultralytics/yolo/utils/metrics.py:514
↓ 1 callersMethodmean_results
(self)
ultralytics/yolo/utils/metrics.py:571
↓ 1 callersFunctionmixup
(im, labels, im2, labels2)
ultralytics/yolo/data/dataloaders/v5augmentations.py:294
← previousnext →201–300 of 814, ranked by callers