MCPcopy Create free account

hub / github.com/Tianxiaomo/pytorch-YOLOv4 / functions

Functions222 in github.com/Tianxiaomo/pytorch-YOLOv4

⨍ Functions 222◇ Types & classes 36
↓ 161 callersFunctionprint
(*args, **kwargs)
tool/tv_reference/utils.py:265
↓ 29 callersMethodmax
(self)
tool/tv_reference/utils.py:59
↓ 11 callersMethod__init__
(self)
models.py:9
↓ 9 callersFunctionload_class_names
(namesfile)
tool/utils.py:161
↓ 8 callersFunctionconvert2cpu
(gpu_matrix)
tool/torch_utils.py:67
↓ 8 callersFunctionplot_boxes_cv2
(img, boxes, savename=None, class_names=None, color=None)
tool/utils.py:99
↓ 7 callersMethod__init__
(self)
tool/darknet2pytorch.py:11
↓ 7 callersFunctiondo_detect
(model, img, conf_thresh, nms_thresh, use_cuda=1)
tool/torch_utils.py:76
↓ 7 callersMethodupdate
(self, **kwargs)
tool/tv_reference/utils.py:150
↓ 6 callersMethodload_weights
(self, weightfile)
tool/darknet2pytorch.py:432
↓ 6 callersFunctiontransform_to_onnx
(weight_file, batch_size, n_classes, IN_IMAGE_H, IN_IMAGE_W)
demo_pytorch2onnx.py:15
↓ 5 callersFunction_true_divide
(dividend, divisor)
tool/utils_iou.py:22
↓ 5 callersMethodprint_network
(self)
tool/darknet2pytorch.py:235
↓ 4 callersFunction_true_divide
(dividend, divisor)
tool/utils_iou_test.py:15
↓ 4 callersFunctionfilter_truth
(bboxes, dx, dy, sx, sy, xd, yd)
dataset.py:184
↓ 3 callersMethodaccumulate
(self)
tool/tv_reference/coco_eval.py:55
↓ 3 callersFunctionbboxes_iou
Calculate the Intersection of Unions (IoUs) between bounding boxes. IoU is calculated as a ratio of area of the intersection and area of the u
tool/utils_iou.py:28
↓ 3 callersFunctionget_color
(c, x, max_val)
tool/utils.py:104
↓ 3 callersFunctionis_dist_avail_and_initialized
()
tool/tv_reference/utils.py:273
↓ 3 callersFunctionparse_cfg
(cfgfile)
tool/config.py:5
↓ 3 callersFunctionpost_processing
(img, conf_thresh, nms_thresh, output)
tool/utils.py:172
↓ 3 callersMethodsummarize
(self)
tool/tv_reference/coco_eval.py:59
↓ 2 callersMethod__str__
(self)
demo_trt.py:69
↓ 2 callersFunctionbbox_iou
(box1, box2, x1y1x2y2=True)
tool/utils.py:22
↓ 2 callersFunctionbboxes_iou
Calculate the Intersection of Unions (IoUs) between bounding boxes. IoU is calculated as a ratio of area of the intersection and area of the u
train.py:41
↓ 2 callersFunctioncompute_aspect_ratios
(dataset, indices=None)
tool/tv_reference/group_by_aspect_ratio.py:162
↓ 2 callersFunctionconvert_to_coco_api
tool/tv_reference/coco_utils.py:146
↓ 2 callersFunctiondetect
(session, image_src, namesfile)
demo_darknet2onnx.py:32
↓ 2 callersFunctionevaluate
(model, data_loader, device)
tool/tv_reference/engine.py:71
↓ 2 callersFunctionget_class_name
(cat)
evaluate_on_coco.py:31
↓ 2 callersFunctionget_dataset
(name, image_set, transform, data_path)
tool/tv_reference/train.py:40
↓ 2 callersFunctionget_region_boxes
(boxes_and_confs)
tool/torch_utils.py:48
↓ 2 callersFunctionget_transform
(train)
tool/tv_reference/train.py:51
↓ 2 callersFunctionget_world_size
()
tool/tv_reference/utils.py:281
↓ 2 callersFunctionload_fc
(buf, start, fc_model)
tool/config.py:247
↓ 2 callersMethodlog_every
(self, iterable, print_freq, header=None)
tool/tv_reference/utils.py:180
↓ 2 callersFunctionprint_cfg
(blocks)
tool/config.py:38
↓ 2 callersFunctionrand_scale
(s)
dataset.py:32
↓ 2 callersFunctionrand_uniform_strong
(min, max)
dataset.py:24
↓ 2 callersMethodsynchronize_between_processes
(self)
tool/tv_reference/coco_eval.py:50
↓ 2 callersMethodupdate
(self, value, n=1)
tool/tv_reference/utils.py:26
↓ 2 callersMethodupdate
(self, predictions)
tool/tv_reference/coco_eval.py:35
↓ 1 callersMethod__getitem__
(self, index)
dataset.py:269
↓ 1 callersFunction_coco_remove_images_without_annotations
(dataset, cat_list=None)
tool/tv_reference/coco_utils.py:106
↓ 1 callersFunction_compute_aspect_ratios_coco_dataset
(dataset, indices=None)
tool/tv_reference/group_by_aspect_ratio.py:131
↓ 1 callersFunction_compute_aspect_ratios_custom_dataset
(dataset, indices=None)
tool/tv_reference/group_by_aspect_ratio.py:120
↓ 1 callersFunction_compute_aspect_ratios_slow
(dataset, indices=None)
tool/tv_reference/group_by_aspect_ratio.py:87
↓ 1 callersFunction_compute_aspect_ratios_subset_dataset
(dataset, indices=None)
tool/tv_reference/group_by_aspect_ratio.py:154
↓ 1 callersFunction_compute_aspect_ratios_voc_dataset
(dataset, indices=None)
tool/tv_reference/group_by_aspect_ratio.py:142
↓ 1 callersFunction_count_visible_keypoints
(anno)
tool/tv_reference/coco_utils.py:110
↓ 1 callersFunction_flip_coco_person_keypoints
(kps, width)
tool/tv_reference/transforms.py:7
↓ 1 callersFunction_get_iou_types
(model)
tool/tv_reference/engine.py:58
↓ 1 callersMethod_get_val_item
dataset.py:389
↓ 1 callersFunction_has_only_empty_bbox
(anno)
tool/tv_reference/coco_utils.py:107
↓ 1 callersFunction_has_valid_annotation
(anno)
tool/tv_reference/coco_utils.py:115
↓ 1 callersFunction_quantize
(x, bins)
tool/tv_reference/group_by_aspect_ratio.py:179
↓ 1 callersFunction_repeat_to_at_least
(iterable, n)
tool/tv_reference/group_by_aspect_ratio.py:17
↓ 1 callersMethodadd_meter
(self, name, meter)
tool/tv_reference/utils.py:177
↓ 1 callersFunctionallocate_buffers
(engine, batch_size)
demo_trt.py:76
↓ 1 callersFunctionarg_parse
Parse arguements to the detect module
tool/camera.py:21
↓ 1 callersFunctionbbox_ious
(boxes1, boxes2, x1y1x2y2=True)
tool/torch_utils.py:16
↓ 1 callersFunctionblend_truth_mosaic
(out_img, img, bboxes, w, h, cut_x, cut_y, i_mixup, left_shift, right_shift, top_shift,
dataset.py:213
↓ 1 callersMethodbuild_target
(self, pred, labels, batchsize, fsize, n_ch, output_id)
train.py:167
↓ 1 callersFunctionbuild_targets
(pred_boxes, target, anchors, num_anchors, num_classes, nH, nW, noobject_scale, object_scale,
tool/region_loss.py:6
↓ 1 callersFunctionconvert_coco_poly_to_mask
(segmentations, height, width)
tool/tv_reference/coco_utils.py:33
↓ 1 callersFunctionconvert_to_xywh
(boxes, fmt='voc')
tool/tv_reference/coco_eval.py:163
↓ 1 callersFunctioncreateIndex
(self)
tool/tv_reference/coco_eval.py:212
↓ 1 callersFunctioncreate_aspect_ratio_groups
(dataset, k=0)
tool/tv_reference/group_by_aspect_ratio.py:186
↓ 1 callersFunctioncreate_common_coco_eval
(coco_eval, img_ids, eval_imgs)
tool/tv_reference/coco_eval.py:194
↓ 1 callersMethodcreate_network
(self, blocks)
tool/darknet2pytorch.py:238
↓ 1 callersFunctiondemo_tensorflow
(tfpb_file="./weight/yolov4.pb", image_path=None, print_sensor_name=False)
demo_tensorflow.py:10
↓ 1 callersFunctiondetect
(context, buffers, image_src, image_size, num_classes)
demo_trt.py:153
↓ 1 callersFunctiondetect_cv2
(cfgfile, weightfile, imgfile)
demo.py:26
↓ 1 callersFunctiondetect_cv2_camera
(cfgfile, weightfile)
demo.py:60
↓ 1 callersFunctiondo_inference
(context, bindings, inputs, outputs, stream)
demo_trt.py:105
↓ 1 callersFunctiondraw_box
(img, bboxes)
dataset.py:236
↓ 1 callersFunctionevaluate
finished, tested
train.py:467
↓ 1 callersFunctionevaluate
Run per image evaluation on given images and store results (a list of dict) in self.evalImgs :return: None
tool/tv_reference/coco_eval.py:310
↓ 1 callersFunctionevaluate_on_coco
(cfg, resFile)
evaluate_on_coco.py:94
↓ 1 callersFunctionfill_truth_detection
(bboxes, num_boxes, classes, flip, dx, dy, sx, sy, net_w, net_h)
dataset.py:47
↓ 1 callersFunctionget_args
(**kwargs)
train.py:531
↓ 1 callersFunctionget_args
(**kwargs)
evaluate_on_coco.py:234
↓ 1 callersFunctionget_args
()
demo.py:141
↓ 1 callersFunctionget_coco
(root, image_set, transforms, mode='instances')
tool/tv_reference/coco_utils.py:234
↓ 1 callersFunctionget_coco_api_from_dataset
(dataset)
tool/tv_reference/coco_utils.py:209
↓ 1 callersFunctionget_date_str
()
train.py:574
↓ 1 callersFunctionget_date_str
()
evaluate_on_coco.py:263
↓ 1 callersFunctionget_engine
(engine_path)
demo_trt.py:145
↓ 1 callersFunctionget_image_id
Convert a string to a integer. Make sure that the images and the `image_id`s are in one-one correspondence. There are already `image_id`s
dataset.py:412
↓ 1 callersFunctionget_rank
()
tool/tv_reference/utils.py:287
↓ 1 callersFunctionimage_data_augmentation
(mat, w, h, pleft, ptop, swidth, sheight, flip, dhue, dsat, dexp, gaussian_noise, blur,
dataset.py:103
↓ 1 callersFunctioninit_logger
log_dir: 日志文件的文件夹路径 mode: 'a', append; 'w', 覆盖原文件写入.
train.py:569
↓ 1 callersFunctioninit_logger
log_dir: 日志文件的文件夹路径 mode: 'a', append; 'w', 覆盖原文件写入.
evaluate_on_coco.py:257
↓ 1 callersFunctionis_main_process
()
tool/tv_reference/utils.py:293
↓ 1 callersFunctionloadRes
Load result file and return a result api object. :param resFile (str) : file name of result file :return: res (obj) : resul
tool/tv_reference/coco_eval.py:247
↓ 1 callersFunctionload_conv
(buf, start, conv_model)
tool/config.py:197
↓ 1 callersFunctionload_conv_bn
(buf, start, conv_model, bn_model)
tool/config.py:216
↓ 1 callersFunctionmain
(engine_path, image_path, image_size)
demo_trt.py:120
↓ 1 callersFunctionmain
(weight_file, image_path, batch_size, n_classes, IN_IMAGE_H, IN_IMAGE_W)
demo_pytorch2onnx.py:66
↓ 1 callersFunctionmain
(cfg_file, namesfile, weight_file, image_path, batch_size)
demo_darknet2onnx.py:13
next →1–100 of 222, ranked by callers