↓ 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
↓ 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 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 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
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,
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,
data_dir=None,
train=True,
img_size=(416, 416),
preproc=None,
dass_det/data/datasets/manga109.py:18
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,
shortcut=True,
expansion=0.5,
dass_det/models/network_blocks.py:81
Method__init__(
self, in_channels, out_channels, kernel_sizes=(5, 9, 13), activation="silu"
)
dass_det/models/network_blocks.py:125
Method__init__(
self,
depth=1.0,
width=1.0,
in_features=("dark3", "dark4", "dark5"),
dass_det/models/yolo_pafpn.py:17
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