MCPcopy Create free account

hub / github.com/barisbatuhan/DASS_Det_Inference / functions

Functions148 in github.com/barisbatuhan/DASS_Det_Inference

↓ 13 callersFunctionpostprocess
(prediction, num_classes, conf_thre=0.7, nms_thre=0.45, class_agnostic=False, helper_prediction=None)
dass_det/utils/boxes.py:84
↓ 6 callersMethod__init__
(self, in_channels, out_channels, ksize=1, stride=1, act="silu")
dass_det/models/network_blocks.py:191
↓ 6 callersFunctionget_aug_params
(value, center=0)
dass_det/data/data_augment.py:203
↓ 6 callersFunctionpreproc
(img, input_size, swap=(2, 0, 1))
dass_det/data/data_augment.py:346
↓ 5 callersMethodmake_group_layer
starts with conv layer then has `num_blocks` `ResLayer`
dass_det/models/darknet.py:59
↓ 3 callersMethod__read_xmls__
(self, annot_path, files, train :bool)
dass_det/data/datasets/comic2k.py:133
↓ 3 callersMethodload_image
(self, index)
dass_det/data/datasets/dcm772.py:92
↓ 2 callersMethod__init__
(self, input_dimension, mosaic=True)
dass_det/data/datasets/datasets_wrapper.py:70
↓ 2 callersFunctionaugment_hsv
(img, hgain=0.015, sgain=0.7, vgain=0.4)
dass_det/data/data_augment.py:171
↓ 2 callersMethodclear
(self)
dass_det/utils/metric.py:96
↓ 2 callersMethodload_anno
(self, index)
dass_det/data/datasets/dcm772.py:57
↓ 2 callersMethodload_image
(self, index)
dass_det/data/datasets/comic2k.py:83
↓ 2 callersMethodload_image
(self, index)
dass_det/data/datasets/ebdtheque.py:88
↓ 2 callersMethodload_image
(self, index)
dass_det/data/datasets/icartoonface.py:80
↓ 2 callersMethodload_image
(self, index)
dass_det/data/datasets/manga109.py:88
↓ 2 callersFunctionxyxy2cxcywh
(bboxes)
dass_det/utils/boxes.py:203
↓ 1 callersMethod__init__
Args: depth (int): depth of darknet used in model, usually use [21, 53] for this param. in_channels (int): number of
dass_det/models/darknet.py:14
↓ 1 callersFunction_mirror
(image, boxes, prob=0.5)
dass_det/data/data_augment.py:311
↓ 1 callersFunction_vertical_mirror
(image, boxes, prob=0.5)
dass_det/data/data_augment.py:319
↓ 1 callersFunctionapply_affine_to_bboxes
(targets, target_size, M, scale)
dass_det/data/data_augment.py:256
↓ 1 callersFunctionconvert_grayscale
(img)
dass_det/data/data_augment.py:165
↓ 1 callersMethoddecode_outputs
(self, outputs, dtype)
dass_det/models/yolo_head.py:126
↓ 1 callersFunctiondraw_ellipse
(img, center, radius_x, radius_y, color=(255, 255, 255))
dass_det/data/data_augment.py:21
↓ 1 callersFunctiondraw_rounded_rectangle
(img, center, width, height, radius=None, color=(255, 255, 255))
dass_det/data/data_augment.py:37
↓ 1 callersMethodextract_components
(self, outputs, info_imgs, ids, select_index :int)
dass_det/evaluators/comic_evaluator.py:205
↓ 1 callersFunctionfuse_conv_and_bn
(conv, bn)
dass_det/utils/model_utils.py:31
↓ 1 callersFunctionget_activation
(name="silu", inplace=True)
dass_det/models/network_blocks.py:17
↓ 1 callersFunctionget_affine_matrix
( target_size, degrees=10, translate=0.1, scales=0.1, shear=10, perp_rotate_prob=0.0,
dass_det/data/data_augment.py:214
↓ 1 callersFunctionget_total_and_free_memory_in_Mb
(cuda_device)
dass_det/utils/metric.py:22
↓ 1 callersFunctionintersect
We resize both tensors to [A,B,2] without new malloc: [A,2] -> [A,1,2] -> [A,B,2] [B,2] -> [1,B,2] -> [A,B,2] Then we compute the area of
dass_det/utils/boxes.py:63
↓ 1 callersFunctionis_parallel
check if model is in parallel mode.
dass_det/utils/ema.py:13
↓ 1 callersMethodload_anno
(self, index)
dass_det/data/datasets/comic2k.py:57
↓ 1 callersMethodload_anno
(self, index)
dass_det/data/datasets/ebdtheque.py:58
↓ 1 callersMethodload_anno
(self, index)
dass_det/data/datasets/icartoonface.py:55
↓ 1 callersMethodload_anno
(self, index)
dass_det/data/datasets/manga109.py:53
↓ 1 callersMethodload_annotations
(self, comic2k_paths, train :bool, filter)
dass_det/data/datasets/comic2k.py:92
↓ 1 callersMethodload_annotations
(self, edb_paths, train :bool)
dass_det/data/datasets/ebdtheque.py:96
↓ 1 callersMethodload_annotations
(self, icf_paths, train :bool)
dass_det/data/datasets/icartoonface.py:88
↓ 1 callersMethodload_annotations
(self, paths, train :bool)
dass_det/data/datasets/dcm772.py:100
↓ 1 callersMethodload_annotations
(self, paths, train :bool)
dass_det/data/datasets/manga109.py:96
↓ 1 callersMethodmake_spp_block
(self, filters_list, in_filters)
dass_det/models/darknet.py:66
↓ 1 callersMethodpull_item
(self, index)
dass_det/data/datasets/comic2k.py:44
↓ 1 callersMethodpull_item
(self, index)
dass_det/data/datasets/ebdtheque.py:45
↓ 1 callersMethodpull_item
(self, index)
dass_det/data/datasets/icartoonface.py:42
↓ 1 callersMethodpull_item
(self, index)
dass_det/data/datasets/dcm772.py:45
↓ 1 callersMethodpull_item
(self, index)
dass_det/data/datasets/manga109.py:40
↓ 1 callersFunctionrandom_crop
(image, min_ratio=0.5)
dass_det/data/data_augment.py:327
Method__call__
(self, image, targets, input_dim)
dass_det/data/data_augment.py:372
Method__call__
(self, img, res, input_size)
dass_det/data/data_augment.py:450
Method__call__
(self, img, input_size)
dass_det/data/data_augment.py:472
Method__call__
(self, img, input_size)
dass_det/data/data_augment.py:508
Method__getitem__
(self, index)
dass_det/data/datasets/comic2k.py:50
Method__getitem__
(self, index)
dass_det/data/datasets/ebdtheque.py:51
Method__getitem__
(self, index)
dass_det/data/datasets/icartoonface.py:48
Method__getitem__
(self, index)
dass_det/data/datasets/dcm772.py:50
Method__getitem__
(self, index)
dass_det/data/datasets/datasets_wrapper.py:41
Method__getitem__
(self, index)
dass_det/data/datasets/manga109.py:46
Method__init__
(self, window_size=50)
dass_det/utils/metric.py:56
Method__init__
(self, window_size=20)
dass_det/utils/metric.py:103
Method__init__
Args: model (nn.Module): model to apply EMA. decay (float): ema decay reate. updates (int): counter of EM
dass_det/utils/ema.py:32
Method__init__
(self, max_labels=50, flip_prob=0.5, hsv_prob=1.0, vertical_flip_prob=0.0)
dass_det/data/data_augment.py:366
Method__init__
(self, swap=(2, 0, 1), legacy=False)
dass_det/data/data_augment.py:445
Method__init__
(self, swap=(2, 0, 1), flip_prob=0.5, hsv_prob=0.7, crop_prob=1.0, min_crop_ratio=0.5, gauss
dass_det/data/data_augment.py:462
Method__init__
(self, swap=(2, 0, 1), flip_prob=0.5)
dass_det/data/data_augment.py:503
Method__init__
( self, data_dir=None, train=True, img_size=(416, 416), preproc=None,
dass_det/data/datasets/comic2k.py:16
Method__init__
( self, data_dir=None, train=True, img_size=(416, 416), preproc=None,
dass_det/data/datasets/ebdtheque.py:25
Method__init__
( self, data_dir=None, train=True, img_size=(416, 416), preproc=None,
dass_det/data/datasets/icartoonface.py:16
Method__init__
( self, data_dir=None, train=True, img_size=(416, 416), preproc=None,
dass_det/data/datasets/dcm772.py:15
Method__init__
(self, datasets)
dass_det/data/datasets/datasets_wrapper.py:13
Method__init__
(self, datasets)
dass_det/data/datasets/datasets_wrapper.py:35
Method__init__
( self, data_dir=None, train=True, img_size=(416, 416), preproc=None,
dass_det/data/datasets/manga109.py:18
Method__init__
Args: dataloader (Dataloader): evaluate dataloader. img_size (int): image size after preprocess. images are resized
dass_det/evaluators/comic_evaluator.py:31
Method__init__
( self, dep_mul, wid_mul, out_features=("dark3", "dark4", "dark5"), de
dass_det/models/darknet.py:98
Method__init__
( self, in_channels, out_channels, ksize, stride, groups=1, bias=False, act="silu" )
dass_det/models/network_blocks.py:32
Method__init__
(self, in_channels, out_channels, ksize, stride=1, act="silu")
dass_det/models/network_blocks.py:60
Method__init__
( self, in_channels, out_channels, shortcut=True, expansion=0.5,
dass_det/models/network_blocks.py:81
Method__init__
(self, in_channels: int)
dass_det/models/network_blocks.py:107
Method__init__
( self, in_channels, out_channels, kernel_sizes=(5, 9, 13), activation="silu" )
dass_det/models/network_blocks.py:125
Method__init__
Args: in_channels (int): input channels. out_channels (int): output channels. n (int): number of Bottlene
dass_det/models/network_blocks.py:150
Method__init__
Args: act (str): activation type of conv. Defalut value: "silu". depthwise (bool): whether apply depthwise conv in co
dass_det/models/yolo_head.py:21
Method__init__
(self, backbone=None, head_stem=None, face_head=None, body_head=None)
dass_det/models/yolox.py:19
Method__init__
( self, depth=1.0, width=1.0, in_features=("dark3", "dark4", "dark5"),
dass_det/models/yolo_pafpn.py:17
Method__init__
(self, width=1.0, in_channels=[256, 512, 1024], act="silu")
dass_det/models/yolo_head_stem.py:11
Method__len__
(self)
dass_det/data/datasets/comic2k.py:40
Method__len__
(self)
dass_det/data/datasets/ebdtheque.py:41
Method__len__
(self)
dass_det/data/datasets/icartoonface.py:38
Method__len__
(self)
dass_det/data/datasets/dcm772.py:42
Method__len__
(self)
dass_det/data/datasets/manga109.py:36
Functionadjust_box_anns
(bbox, scale_ratio, padw, padh, w_max, h_max)
dass_det/utils/boxes.py:191
Methodarrange_files
(self)
dass_det/data/datasets/datasets_wrapper.py:75
Methodavg
(self)
dass_det/utils/metric.py:72
Functionbboxes_iou
(bboxes_a, bboxes_b, xyxy=True)
dass_det/utils/boxes.py:152
Functionbox_candidates
(box1, box2, wh_thr=2, ar_thr=20, area_thr=0.2)
dass_det/data/data_augment.py:187
Functioncheck_center_boxa_in_boxb
Checks if the center of box_a is in box_b. Return: (tensor) int tensor of zeros and ones.
dass_det/utils/boxes.py:35
Methodclear_meters
(self)
dass_det/utils/metric.py:123
Functioncxcywh2xyxy
(bboxes)
dass_det/utils/boxes.py:210
Functiondefault_replace_func
(replaced_module_type, new_module_type)
dass_det/utils/model_utils.py:91
Functiondraw_speech_balloon
(orig_img, ellipse_prob=0.7, max_balloons :int=2, add_noise :bool=True, horizontal_prob=0.75, downwards_prob=0
dass_det/data/data_augment.py:106
Methodevaluate
( self, model, distributed=False, half=False, trt_file=None, d
dass_det/evaluators/comic_evaluator.py:52
Functionfilter_box
output: (N, 5+class) shape
dass_det/utils/boxes.py:25
next →1–100 of 148, ranked by callers