Functions — github.com/Tianxiaomo/pytorch-YOLOv4

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

Functions222 in github.com/Tianxiaomo/pytorch-YOLOv4

⨍ Functions 222◇ Types & classes 36

↓ 1 callersFunctionmain

(args)

tool/tv_reference/train.py:59

↓ 1 callersFunctionmerge

(img_ids, eval_imgs)

tool/tv_reference/coco_eval.py:172

↓ 1 callersFunctionnms_cpu

(boxes, confs, nms_thresh=0.5, min_mode=False)

tool/utils.py:61

↓ 1 callersMethodprepare

(self, predictions, iou_type)

tool/tv_reference/coco_eval.py:64

↓ 1 callersMethodprepare_for_coco_detection

(self, predictions)

tool/tv_reference/coco_eval.py:74

↓ 1 callersMethodprepare_for_coco_keypoint

(self, predictions)

tool/tv_reference/coco_eval.py:136

↓ 1 callersMethodprepare_for_coco_segmentation

(self, predictions)

tool/tv_reference/coco_eval.py:101

↓ 1 callersFunctionrect_intersection

(a, b)

dataset.py:94

↓ 1 callersFunctionsetup_for_distributed

This function disables printing when not in master process

tool/tv_reference/utils.py:258

↓ 1 callersMethodsynchronize_between_processes

(self)

tool/tv_reference/utils.py:173

↓ 1 callersFunctiontest

(model, annotations, cfg)

evaluate_on_coco.py:160

↓ 1 callersFunctiontrain

(model, device, config, epochs=5, batch_size=1, save_cp=True, log_step=20, img_scale=0.5)

train.py:291

↓ 1 callersFunctiontrain_one_epoch

(model, optimizer, data_loader, device, epoch, print_freq)

tool/tv_reference/engine.py:13

↓ 1 callersFunctiontransform_to_onnx

(cfgfile, weightfile, batch_size=1, onnx_file_name=None)

tool/darknet2onnx.py:6

↓ 1 callersFunctiontransform_to_tensorflow

(onnx_input_path, pb_output_path)

tool/onnx2tensorflow.py:13

↓ 1 callersFunctionyolo_forward_dynamic

(output, conf_thresh, num_classes, anchors, num_anchors, scale_x_y, only_objectness=1,

tool/yolo_layer.py:148

(self, image, target)

tool/tv_reference/transforms.py:21

Method__call__

(self, image, target)

tool/tv_reference/transforms.py:31

Method__call__

(self, image, target)

tool/tv_reference/transforms.py:48

Method__call__

(self, image, target)

tool/tv_reference/coco_utils.py:20

Method__call__

(self, image, target)

tool/tv_reference/coco_utils.py:51

Method__getattr__

(self, attr)

tool/tv_reference/utils.py:157

Method__getitem__

(self, idx)

tool/tv_reference/coco_utils.py:225

Method__init__

(self, n_classes=80, n_anchors=3, device=None, batch=2)

train.py:131

Method__init__

(self, host_mem, device_mem)

(self, in_channels, out_channels, kernel_size, stride, activation, bn=True, bias=False)

models.py:40

Method__init__

(self, ch, nblocks=1, shortcut=True)

(self, inference=False)

models.py:240

Method__init__

(self, output_ch, n_classes, inference=False)

models.py:324

Method__init__

(self, yolov4conv137weight=None, n_classes=80, inference=False)

models.py:410

Method__init__

(self, label_path, cfg, train=True)

dataset.py:243

Method__init__

(self, anchor_mask=[], num_classes=0, anchors=[], num_anchors=1, stride=32, model_out=False)

tool/yolo_layer.py:295

Method__init__

(self, num_classes=0, anchors=[], num_anchors=1)

tool/region_loss.py:105

Method__init__

(self, size=2, stride=1)

tool/darknet2pytorch.py:20

Method__init__

(self, stride=2)

tool/darknet2pytorch.py:53

Method__init__

(self, stride)

tool/darknet2pytorch.py:68

Method__init__

(self, stride=2)

tool/darknet2pytorch.py:80

Method__init__

(self)

tool/darknet2pytorch.py:103

Method__init__

(self)

tool/darknet2pytorch.py:118

Method__init__

(self, cfgfile, inference=False)

tool/darknet2pytorch.py:127

Method__init__

(self, transforms)

tool/tv_reference/transforms.py:18

Method__init__

(self, prob)

tool/tv_reference/transforms.py:28

Method__init__

(self, window_size=20, fmt=None)

tool/tv_reference/utils.py:18

Method__init__

(self, delimiter="\t")

tool/tv_reference/utils.py:146

Method__init__

(self, coco_gt, iou_types, bbox_fmt='coco')

tool/tv_reference/coco_eval.py:20

Method__init__

(self, categories, remap=True)

tool/tv_reference/coco_utils.py:16

Method__init__

(self, img_folder, ann_file, transforms)

tool/tv_reference/coco_utils.py:221

Method__init__

(self, sampler, group_ids, batch_size)

tool/tv_reference/group_by_aspect_ratio.py:37

Method__init__

(self, indices)

tool/tv_reference/group_by_aspect_ratio.py:96

Method__iter__

(self)

tool/tv_reference/group_by_aspect_ratio.py:47

Method__iter__

(self)

tool/tv_reference/group_by_aspect_ratio.py:99

tool/tv_reference/group_by_aspect_ratio.py:83

Method__len__

(self)

tool/tv_reference/group_by_aspect_ratio.py:102

tool/tv_reference/utils.py:66

Method__str__

(self)

tool/tv_reference/utils.py:165

Function_get_date_str

()

train.py:604

Functionall_gather

Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]

tool/tv_reference/utils.py:75

Methodavg

(self)

tool/tv_reference/utils.py:50

Functionbboxes_ciou

(bboxes_a, bboxes_b, fmt='voc')

tool/utils_iou.py:203

Functionbboxes_diou

(bboxes_a, bboxes_b, fmt='voc')

tool/utils_iou.py:199

Functionbboxes_giou

(bboxes_a, bboxes_b, fmt='voc')

tool/utils_iou.py:195

Functionbboxes_iou_test

test function for the bboxes_iou function in `train_acne.py`, with message printing and plot

tool/utils_iou_test.py:22

Functionburnin_schedule

tool/tv_reference/utils.py:235

Functionconvert2cpu_long

(gpu_matrix)

tool/torch_utils.py:71

Functionconvert_cat_id_and_reorientate_bbox

(single_annotation)

evaluate_on_coco.py:53

Functiondetect_skimage

(cfgfile, weightfile, imgfile)

demo.py:107

Functionf

(x)

tool/tv_reference/utils.py:241

Functionfind_sample_data

Parses sample arguments. Args: description (str): Description of the sample. subfolder (str): The subfolder containing data r

demo_trt.py:23

Methodforward

(self, xin, labels=None)

(self, x, target_size, inference=False)

(self, input, downsample4, downsample3, inference=False)

models.py:280

Methodforward

(self, input1, input2, input3)

(self, output, target=None)

tool/yolo_layer.py:313

Methodforward

(self, output, target)

tool/region_loss.py:118

Methodforward

(self, x)

tool/darknet2pytorch.py:14

Methodforward

darknet output_size = (input_size + p - k) / s +1 p : padding = k - 1 k : size s : stride torch output_size =

tool/darknet2pytorch.py:25

Methodforward

(self, x)

tool/darknet2pytorch.py:57

Methodforward

(self, x)

tool/darknet2pytorch.py:72

Methodforward

(self, x)

tool/darknet2pytorch.py:84

Methodforward

(self, x)

tool/darknet2pytorch.py:106

← previous next →101–200 of 222, ranked by callers