File size: 889 Bytes
3e99b05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
detrex.modeling
##############################

backbone
------------------------------
.. currentmodule:: detrex.modeling
.. automodule:: detrex.modeling.backbone
    :member-order: bysource
    :members:
        ResNet,
        make_stage,
        ConvNeXt,
        FocalNet,
        TimmBackbone,
        TorchvisionBackbone,

neck
------------------------------
.. currentmodule:: detrex.modeling
.. automodule:: detrex.modeling.neck
    :member-order: bysource
    :members:
        ChannelMapper,


matcher
------------------------------
.. currentmodule:: detrex.modeling
.. automodule:: detrex.modeling.matcher
    :member-order: bysource
    :members:
        HungarianMatcher,


losses
------------------------------
.. currentmodule:: detrex.modeling
.. automodule:: detrex.modeling.losses
    :member-order: bysource
    :members:
        sigmoid_focal_loss,
        dice_loss,