MCPcopy Create free account

hub / github.com/RizwanMunawar/yolov7-segmentation / functions

Functions646 in github.com/RizwanMunawar/yolov7-segmentation

↓ 172 callersMethodinfo
(self, verbose=False, img_size=640)
models/yolo.py:241
↓ 47 callersFunctioncolorstr
(*input)
utils/general.py:661
↓ 32 callersMethod__init__
(self, k=2)
models/common.py:63
↓ 31 callersMethodrun
Loop through the registered actions and fire all callbacks on main thread Args: hook: The name of the hook to check,
utils/callbacks.py:60
↓ 27 callersMethodtime
(self)
utils/general.py:157
↓ 22 callersMethodtolist
(self)
models/common.py:819
↓ 17 callersMethod__init__
(self, w=None)
models/tf.py:40
↓ 17 callersFunctioncheck_requirements
(requirements=ROOT / 'requirements.txt', exclude=(), install=True, cmds=())
utils/general.py:372
↓ 16 callersMethodplot
(self, normalize=True, save_dir='', names=())
utils/metrics.py:190
↓ 15 callersFunctioncheck_version
(current='0.0.0', minimum='0.0.0', name='version ', pinned=False, hard=False, verbose=False)
utils/general.py:359
↓ 14 callersFunctionxywh2xyxy
(x)
utils/general.py:735
↓ 13 callersMethodsave
(self, labels=True, save_dir='runs/detect/exp')
models/common.py:797
↓ 12 callersMethodsave
Save the ONNX model to the given location. Args: output_path: Path pointing to the location where to write
utils/add_nms.py:60
↓ 12 callersFunctionselect_device
(device='', batch_size=0, newline=True)
utils/torch_utils.py:107
↓ 11 callersFunctionde_parallel
(model)
utils/torch_utils.py:206
↓ 11 callersMethodlog
save the metrics to the logging dictionary arguments: log_dict (Dict) -- metrics/media to be logged in current step
utils/loggers/wandb/wandb_utils.py:515
↓ 11 callersFunctionxyxy2xywh
(x)
utils/general.py:725
↓ 10 callersFunctioncheck_file
(file, suffix='')
utils/general.py:451
↓ 10 callersFunctioncheck_yaml
(file, suffix=('.yaml', '.yml'))
utils/general.py:446
↓ 10 callersFunctionfitness
(x)
utils/metrics.py:17
↓ 10 callersFunctionincrement_path
(path, exist_ok=False, sep='', mkdir=False)
utils/general.py:1145
↓ 9 callersFunctioncheck_dataset
(data, autodownload=True)
utils/general.py:489
↓ 9 callersFunctionprint_args
(args: Optional[dict] = None, show_file=True, show_func=False)
utils/general.py:203
↓ 9 callersMethodrectangle
(self, xy, fill=None, outline=None, width=1)
utils/plots.py:141
↓ 9 callersFunctionxywhn2xyxy
(x, w=640, h=640, padw=0, padh=0)
utils/general.py:745
↓ 8 callersFunctiontorch_distributed_zero_first
(local_rank: int)
utils/torch_utils.py:88
↓ 8 callersMethodupdate
Parameters: 'dets' - a numpy array of detection in the format [[x1, y1, x2, y2, score], [x1,y1,x2,y2,score],...]
segment/sort_count.py:253
↓ 7 callersMethod__init__
(self, nc=80, anchors=(), ch=(), inplace=True)
models/yolo.py:44
↓ 7 callersFunctionletterbox
(img, new_shape=(640, 640), color=(114, 114, 114), auto=True, scaleFill=False, scaleup=True, stride=32)
utils/datasets.py:984
↓ 7 callersMethodlog
(self, log_dict)
utils/wandb_logging/wandb_utils.py:285
↓ 7 callersFunctionscale_coords
(img1_shape, coords, img0_shape, ratio_pad=None)
utils/general.py:802
↓ 7 callersFunctionxyn2xy
(x, w=640, h=640, padw=0, padh=0)
utils/general.py:767
↓ 6 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/metrics.py:270
↓ 6 callersFunctioncheck_img_size
(imgsz, s=32, floor=0)
utils/general.py:408
↓ 6 callersFunctioncreate_dataloader
(path, imgsz, batch_size, stride,
utils/dataloaders.py:101
↓ 6 callersMethodfromarray
(self, im)
utils/plots.py:152
↓ 6 callersFunctionmake_divisible
(x, divisor)
utils/general.py:644
↓ 6 callersMethodupdate
Args: results: tuple(p, r, ap, f1, ap_class)
utils/segment/metrics.py:137
↓ 5 callersFunctionattempt_load
(weights, device=None, inplace=True, fuse=True)
models/experimental.py:73
↓ 5 callersMethodbox_label
(self, box, label='', color=(128, 128, 128), txt_color=(255, 255, 255))
utils/plots.py:85
↓ 5 callersMethoddisplay
(self, pprint=False, show=False, save=False, crop=False, render=False, labels=True, save_dir=Path(''))
models/common.py:747
↓ 5 callersFunctionimg2label_paths
(img_paths)
utils/dataloaders.py:403
↓ 5 callersFunctionimg2label_paths
(img_paths)
utils/datasets.py:347
↓ 5 callersFunctionletterbox
(im, new_shape=(640, 640), color=(114, 114, 114), auto=True, scaleFill=False, scaleup=True, stride=32)
utils/augmentations.py:109
↓ 5 callersFunctionload_image
(self, index)
utils/datasets.py:666
↓ 5 callersMethodload_image
(self, i)
utils/dataloaders.py:672
↓ 5 callersMethodpredict
(self, inputs, tf_nms=False, agnostic_nms=False,
models/tf.py:421
↓ 5 callersFunctionremove_prefix
(from_string, prefix=WANDB_ARTIFACT_PREFIX)
utils/wandb_logging/wandb_utils.py:23
↓ 5 callersFunctionremove_prefix
(from_string, prefix=WANDB_ARTIFACT_PREFIX)
utils/loggers/wandb/wandb_utils.py:32
↓ 5 callersFunctiontime_sync
()
utils/torch_utils.py:143
↓ 4 callersFunctionattempt_download
(file, repo='ultralytics/yolov5', release='v6.2')
utils/downloads.py:57
↓ 4 callersFunctionautopad
(k, p=None)
models/common.py:32
↓ 4 callersFunctioncheck_suffix
(file='yolov5s.pt', suffix=('.pt',), msg='')
utils/general.py:435
↓ 4 callersFunctionis_colab
()
utils/general.py:70
↓ 4 callersFunctionmetric
(k)
utils/autoanchor.py:35
↓ 4 callersFunctionnon_max_suppression
Non-Maximum Suppression (NMS) on inference results to reject overlapping detections Returns: list of detections, on (n,6) tensor per
utils/general.py:945
↓ 4 callersMethodprocess_batch
Return intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in (x1, y1, x2, y2) format.
utils/metrics.py:134
↓ 4 callersFunctionyaml_save
(file='data.yaml', data={})
utils/general.py:583
↓ 3 callersMethod_forward_once
(self, x, profile=False, visualize=False)
models/yolo.py:205
↓ 3 callersFunctionap_per_class
Compute the average precision, given the recall and precision curves. Source: https://github.com/rafaelpadilla/Object-Detection-Metrics. #
utils/metrics.py:31
↓ 3 callersMethodbackward
(ctx, grad_output)
utils/activations.py:43
↓ 3 callersFunctionbbox_ioa
(box1, box2)
utils/datasets.py:1114
↓ 3 callersMethodcalculate_centroid
(self,bbox)
segment/sort_count.py:140
↓ 3 callersFunctioncheck_online
()
utils/general.py:307
↓ 3 callersFunctionclip_coords
(boxes, shape)
utils/general.py:818
↓ 3 callersFunctioncopy_paste
(img, labels, segments, probability=0.5)
utils/datasets.py:894
↓ 3 callersFunctioncrop
"Crop" predicted masks by zeroing out everything not in the predicted bbox. Vectorized by Chong (thanks Chong). Args: - mas
utils/segment/general.py:6
↓ 3 callersFunctionexport_onnx
(model, im, file, opset, train, dynamic, simplify, prefix=colorstr('ONNX:'))
export.py:127
↓ 3 callersFunctionfile_size
(path)
utils/general.py:295
↓ 3 callersFunctionis_ascii
(s='')
utils/general.py:59
↓ 3 callersMethodlog_images
(self, files, name='Images', epoch=0)
utils/loggers/__init__.py:278
↓ 3 callersMethodlog_metrics
(self, metrics, epoch)
utils/loggers/__init__.py:262
↓ 3 callersMethodlog_model
(self, model_path, epoch=0, metadata={})
utils/loggers/__init__.py:295
↓ 3 callersFunctionparse_opt
(known=False)
train.py:415
↓ 3 callersFunctionparse_opt
()
utils/benchmarks.py:134
↓ 3 callersFunctionplot_images
(images, targets, paths=None, fname='images.jpg', names=None)
utils/plots.py:224
↓ 3 callersFunctionplot_images_and_masks
(images, targets, masks, paths=None, fname='images.jpg', names=None)
utils/segment/plots.py:55
↓ 3 callersFunctionplot_mc_curve
(px, py, save_dir=Path('mc_curve.png'), names=(), xlabel='Confidence', ylabel='Metric')
utils/metrics.py:348
↓ 3 callersFunctionprint_results
(k, verbose=True)
utils/autoanchor.py:100
↓ 3 callersFunctionrandom_perspective
(im, targets=(), segments=(), degrees=
utils/augmentations.py:142
↓ 3 callersFunctionrandom_perspective
(img, targets=(), segments=(), degrees=10, translate=.1, scale=.1, shear=10, perspective=0.0,
utils/datasets.py:1017
↓ 3 callersFunctionresample_segments
(segments, n=1000)
utils/general.py:792
↓ 3 callersFunctionrun
( data, weights=None, # model.pt path(s) batch_size=32, # batch size imgsz=6
val.py:97
↓ 3 callersFunctionrun
( data, weights=None, # model.pt path(s) batch_size=32, # batch size img
segment/val.py:128
↓ 3 callersFunctionsegment2box
(segment, width=640, height=640)
utils/general.py:775
↓ 3 callersFunctionstrip_optimizer
(f='best.pt', s='')
utils/general.py:1054
↓ 3 callersMethodtext
(self, xy, text, txt_color=(255, 255, 255), anchor='top')
utils/plots.py:145
↓ 3 callersFunctiontrain
(hyp, opt, device, callbacks)
train.py:50
↓ 3 callersMethodwarmup
(self, imgsz=(1, 3, 640, 640))
models/common.py:603
↓ 3 callersFunctionyaml_load
(file='data.yaml')
utils/general.py:577
↓ 2 callersMethod__init__
(self, loss_fcn, gamma=1.5, alpha=0.25)
utils/loss.py:37
↓ 2 callersMethod__init__
(self, path, img_size=640, stride=32, auto=True, transforms=None)
utils/dataloaders.py:191
↓ 2 callersMethod__init__
(self, c1, k=3)
utils/activations.py:55
↓ 2 callersMethod__init__
(self, n, weight=False)
models/experimental.py:16
↓ 2 callersFunctionactivations
(act=nn.SiLU)
models/tf.py:508
↓ 2 callersFunctionall_logging_disabled
source - https://gist.github.com/simon-weber/7853144 A context manager that will prevent any logging messages triggered during the body from bei
utils/loggers/wandb/wandb_utils.py:573
↓ 2 callersFunctionamp_allclose
(model, im)
utils/general.py:553
↓ 2 callersFunctionanchor_fitness
(k)
utils/autoanchor.py:96
↓ 2 callersFunctionaugment_hsv
(img, hgain=0.5, sgain=0.5, vgain=0.5)
utils/datasets.py:683
↓ 2 callersFunctionbbox_ioa
Returns the intersection over box2 area given box1, box2. Boxes are x1y1x2y2 box1: np.array of shape(4) box2: np.array of shape
utils/metrics.py:291
next →1–100 of 646, ranked by callers