Method__init__(self, max_labels=50, flip_prob=0.5, hsv_prob=1.0, legacy=False, d_rate=1/4)
unicorn/data/data_augment.py:692
Method__init__(self, max_labels=50, flip_prob=0.5, hsv_prob=1.0, legacy=False, d_rate=1/4)
unicorn/data/data_augment.py:793
Method__init__(
self,
img_size=(416, 416),
preproc=None,
min_sz=0,
box_only=False
unicorn/data/datasets/youtube_vos.py:18
Method__init__(self, img_size, datasets, p_datasets, samples_per_epoch, mode="joint", fix=False, fix_id=None)
unicorn/data/datasets/omni_data.py:40
Method__init__(
self,
data_dir,
image_sets=[("2007", "trainval"), ("2012", "trainval")],
img
unicorn/data/datasets/voc.py:102
Method__init__(self, d_model=256, nhead=8,
num_encoder_layers=6, dim_feedforward=1024, dropout=0.1,
unicorn/models/deformable_transformer.py:22
Method__init__(self,
d_model=256, d_ffn=1024,
dropout=0.1, activation="relu",
unicorn/models/deformable_transformer.py:93
Method__init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
activation="relu", normalize_before
unicorn/models/transformer_encoder.py:51
Method__init__(
self,
dep_mul,
wid_mul,
out_features=("dark3", "dark4", "dark5"),
de
unicorn/models/backbone/darknet.py:98
Method__init__(
self, in_channels, out_channels, ksize, stride, groups=1, bias=False, act="silu"
)
unicorn/models/backbone/network_blocks.py:32
Method__init__(self, in_channels, out_channels, ksize, stride=1, act="silu")
unicorn/models/backbone/network_blocks.py:60
Method__init__(
self,
in_channels,
out_channels,
shortcut=True,
expansion=0.5,
unicorn/models/backbone/network_blocks.py:81
Method__init__(
self, in_channels, out_channels, kernel_sizes=(5, 9, 13), activation="silu"
)
unicorn/models/backbone/network_blocks.py:125
Method__init__(self, in_chans=3, depths=[3, 3, 9, 3], dims=[96, 192, 384, 768],
drop_path_rate=0., layer_s
unicorn/models/backbone/convnext.py:71
Method__init__(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)
unicorn/models/backbone/swin_transformer.py:85
Method__init__(self, dim, num_heads, window_size=7, shift_size=0,
mlp_ratio=4., qkv_bias=True, qk_scale=Non
unicorn/models/backbone/swin_transformer.py:172
Method__init__(
self,
depth=1.0,
width=1.0,
in_features=("dark3", "dark4", "dark5"),
unicorn/models/backbone/yolo_pafpn_new.py:20
Method__init__(self, inplanes, planes, stride=1, downsample=None, groups=1,
base_width=64, dilation=1, norm
unicorn/models/backbone/resnet.py:41
Method__init__(self, inplanes, planes, stride=1, downsample=None, groups=1,
base_width=64, dilation=1, norm
unicorn/models/backbone/resnet.py:87
Method__init__(self, cfg, in_channels=(192, 384, 768), sem_loss_on=True, use_raft=False, up_rate=8)
unicorn/models/condinst/mask_branch.py:18