MCPcopy Create free account

hub / github.com/WongKinYiu/PyTorch_YOLOv4 / functions

Functions262 in github.com/WongKinYiu/PyTorch_YOLOv4

↓ 17 callersMethodinfo
(self, verbose=False)
models/models.py:563
↓ 11 callersMethod__init__
(self)
utils/layers.py:334
↓ 9 callersFunctionfitness
(k)
utils/utils.py:718
↓ 7 callersFunctionxyxy2xywh
(x)
utils/general.py:143
↓ 6 callersFunctionletterbox
(img, new_shape=(640, 640), color=(114, 114, 114), auto=True, scaleFill=False, scaleup=True, auto_size=32)
utils/datasets.py:1107
↓ 6 callersFunctionload_image
(self, index)
utils/datasets.py:922
↓ 6 callersFunctiontime_synchronized
()
utils/torch_utils.py:68
↓ 6 callersFunctionxywh2xyxy
(x)
utils/general.py:153
↓ 4 callersMethod_get_x_q
(self, x, q, N)
utils/layers.py:308
↓ 4 callersFunctioncheck_file
(file)
utils/general.py:67
↓ 4 callersFunctionget_hash
(files)
utils/datasets.py:40
↓ 4 callersFunctionmetric
(k)
utils/autoanchor.py:29
↓ 4 callersFunctionrandom_perspective
(img, targets=(), degrees=10, translate=.1, scale=.1, shear=10, perspective=0.0, border=(0, 0))
utils/datasets.py:1140
↓ 4 callersFunctionscale_coords
(img1_shape, coords, img0_shape, ratio_pad=None)
utils/general.py:163
↓ 4 callersFunctiontorch_distributed_zero_first
Decorator to make all processes in distributed training wait for each local_master to do something.
utils/torch_utils.py:20
↓ 3 callersFunctionattempt_download
(weights)
utils/google_utils.py:19
↓ 3 callersFunctioncreate_dataloader
(path, imgsz, batch_size, stride, opt, hyp=None, augment=False, cache=False, pad=0.0, rect=False,
utils/datasets.py:60
↓ 3 callersFunctioncreate_folder
(path='./new')
utils/datasets.py:1281
↓ 3 callersMethodcreate_grids
(self, ng=(13, 13), device='cpu')
models/models.py:279
↓ 3 callersMethodcreate_grids
(self, ng=(13, 13), device='cpu')
models/models.py:372
↓ 3 callersFunctionhist2d
(x, y, n=100)
utils/plots.py:34
↓ 3 callersFunctionis_parallel
(model)
utils/torch_utils.py:73
↓ 3 callersFunctionplot_images
(images, targets, paths=None, fname='images.jpg', names=None, max_size=640, max_subplots=16)
utils/plots.py:111
↓ 3 callersFunctionprint_results
(k)
utils/autoanchor.py:81
↓ 3 callersFunctionprint_results
(k)
utils/utils.py:706
↓ 3 callersFunctionselect_device
(device='', batch_size=None)
utils/torch_utils.py:42
↓ 3 callersMethodstep
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
utils/adabound.py:51
↓ 3 callersFunctionwh_iou
(wh1, wh2)
utils/utils.py:330
↓ 3 callersFunctionxywh2xyxy
(x)
utils/utils.py:113
↓ 2 callersMethod__init__
(self, cfg, img_size=(416, 416), verbose=False)
models/models.py:448
↓ 2 callersFunctionanchor_fitness
(k)
utils/autoanchor.py:77
↓ 2 callersFunctionaugment_hsv
(img, hgain=0.5, sgain=0.5, vgain=0.5)
utils/datasets.py:939
↓ 2 callersFunctionbox_area
(box)
utils/general.py:260
↓ 2 callersFunctionbox_area
(box)
utils/utils.py:318
↓ 2 callersFunctionbox_iou
Return intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in (x1, y1, x2, y2) format. Arguments:
utils/general.py:247
↓ 2 callersMethodcache_labels
(self, path='labels.cache3')
utils/datasets.py:511
↓ 2 callersMethodcache_labels
(self, path='labels.cache3')
utils/datasets.py:794
↓ 2 callersFunctioncheck_dataset
(dict)
utils/general.py:78
↓ 2 callersFunctioncheck_img_size
(img_size, s=32)
utils/general.py:59
↓ 2 callersFunctionclip_coords
(boxes, img_shape)
utils/general.py:179
↓ 2 callersFunctioncompute_loss
(p, targets, model)
utils/loss.py:62
↓ 2 callersFunctiondetect
(save_img=False)
detect.py:30
↓ 2 callersFunctionexif_size
(img)
utils/datasets.py:45
↓ 2 callersFunctionfitness
(x)
utils/metrics.py:7
↓ 2 callersMethodforward_once
(self, x, augment=False, verbose=False)
models/models.py:491
↓ 2 callersMethodfuse
(self)
models/models.py:546
↓ 2 callersFunctionincrement_path
(path, exist_ok=True, sep='')
utils/general.py:438
↓ 2 callersFunctionload_darknet_weights
(self, weights, cutoff=-1)
models/models.py:571
↓ 2 callersFunctionload_mosaic
(self, index)
utils/datasets.py:958
↓ 2 callersFunctionload_mosaic9
(self, index)
utils/datasets.py:1016
↓ 2 callersMethodnew_video
(self, path)
utils/datasets.py:216
↓ 2 callersFunctionnon_max_suppression
Performs Non-Maximum Suppression (NMS) on inference results Returns: detections with shape: nx6 (x1, y1, x2, y2, conf, cls)
utils/general.py:280
↓ 2 callersFunctionplot_one_box
(x, img, color=None, label=None, line_thickness=None)
utils/plots.py:54
↓ 2 callersFunctionset_logging
(rank=-1)
utils/general.py:33
↓ 2 callersFunctionstrip_optimizer
(f='weights/best.pt', s='')
utils/general.py:358
↓ 2 callersFunctiontest
(data, weights=None, batch_size=16, imgsz=640, conf_thres=0.001,
test.py:30
↓ 2 callersFunctiontrain
(hyp, opt, device, tb_writer=None, wandb=None)
train.py:44
↓ 2 callersMethodupdate
(self, model)
utils/torch_utils.py:226
↓ 2 callersFunctionxyxy2xywh
(x)
utils/utils.py:103
↓ 1 callersMethod__init__
(self, loss_fcn, gamma=1.5, alpha=0.25)
utils/loss.py:34
↓ 1 callersMethod__init__
(self, path, img_size=640, auto_size=32)
utils/datasets.py:145
↓ 1 callersMethod__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), final_lr=0.1, gamma=1e-3, eps=1e-8, weight_decay=
utils/adabound.py:26
↓ 1 callersMethod__iter__
(self)
utils/datasets.py:174
↓ 1 callersMethod__setstate__
(self, state)
utils/adabound.py:46
↓ 1 callersMethod_get_p
(self, offset, dtype)
utils/layers.py:298
↓ 1 callersMethod_get_p_0
(self, h, w, N, dtype)
utils/layers.py:288
↓ 1 callersMethod_get_p_n
(self, N, dtype)
utils/layers.py:278
↓ 1 callersMethod_reshape_x_offset
(x_offset, ks)
utils/layers.py:325
↓ 1 callersFunctionap_per_class
Compute the average precision, given the recall and precision curves. Source: https://github.com/rafaelpadilla/Object-Detection-Metrics. # Ar
utils/metrics.py:43
↓ 1 callersFunctionapply_classifier
(x, model, img, im0)
utils/general.py:403
↓ 1 callersMethodbackward
(ctx, grad_output)
utils/layers.py:165
↓ 1 callersFunctionbbox_ioa
(box1, box2)
utils/datasets.py:1239
↓ 1 callersFunctionbbox_iou
(box1, box2, x1y1x2y2=True, GIoU=False, DIoU=False, CIoU=False, EIoU=False, ECIoU=False, eps=1e-9)
utils/general.py:187
↓ 1 callersFunctionbbox_iou
(box1, box2, x1y1x2y2=True, GIoU=False, DIoU=False, CIoU=False)
utils/utils.py:259
↓ 1 callersFunctionbox_candidates
(box1, box2, wh_thr=2, ar_thr=20, area_thr=0.1)
utils/datasets.py:1227
↓ 1 callersFunctionbox_iou
Return intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in (x1, y1, x2, y2) format. Arguments:
utils/utils.py:305
↓ 1 callersFunctionbuild_targets
(p, targets, model)
utils/loss.py:127
↓ 1 callersFunctionbuild_targets
(p, targets, model)
utils/utils.py:440
↓ 1 callersFunctionbutter_lowpass
(cutoff, fs, order)
utils/plots.py:45
↓ 1 callersFunctioncheck_anchor_order
(m)
utils/autoanchor.py:10
↓ 1 callersFunctioncheck_git_status
()
utils/general.py:51
↓ 1 callersFunctionclip_coords
(boxes, img_shape)
utils/utils.py:159
↓ 1 callersFunctioncoco80_to_coco91_class
()
utils/general.py:131
↓ 1 callersFunctioncolor_list
()
utils/plots.py:26
↓ 1 callersFunctioncompute_ap
Compute the average precision, given the recall and precision curves. Source: https://github.com/rbgirshick/py-faster-rcnn. # Arguments
utils/utils.py:230
↓ 1 callersFunctioncompute_ap
Compute the average precision, given the recall and precision curves. Source: https://github.com/rbgirshick/py-faster-rcnn. # Arguments
utils/metrics.py:114
↓ 1 callersFunctioncopy_attr
(a, b, include=(), exclude=())
utils/torch_utils.py:197
↓ 1 callersFunctioncreate_modules
(module_defs, img_size, cfg)
models/models.py:9
↓ 1 callersFunctionfitness_ap
(x)
utils/metrics.py:31
↓ 1 callersFunctionfitness_ap50
(x)
utils/metrics.py:25
↓ 1 callersFunctionfitness_f
(x)
utils/metrics.py:37
↓ 1 callersFunctionfitness_p
(x)
utils/metrics.py:13
↓ 1 callersFunctionfitness_r
(x)
utils/metrics.py:19
↓ 1 callersFunctiongdrive_download
(id='1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', name='coco128.zip')
utils/google_utils.py:55
↓ 1 callersFunctionget_latest_run
(search_dir='.')
utils/general.py:45
↓ 1 callersFunctionget_token
(cookie="./cookie")
utils/google_utils.py:89
↓ 1 callersFunctionget_yolo_layers
(model)
models/models.py:567
↓ 1 callersFunctiongsutil_getsize
(url='')
utils/google_utils.py:13
↓ 1 callersFunctionhex2rgb
(h)
utils/plots.py:28
↓ 1 callersFunctioninit_seeds
(seed=0)
utils/general.py:39
next →1–100 of 262, ranked by callers