39 lines
1.0 KiB
ReStructuredText
39 lines
1.0 KiB
ReStructuredText
|
torchreid.models
|
||
|
|
=================
|
||
|
|
|
||
|
|
Interface
|
||
|
|
---------
|
||
|
|
|
||
|
|
.. automodule:: torchreid.models.__init__
|
||
|
|
:members:
|
||
|
|
|
||
|
|
|
||
|
|
ImageNet Classification Models
|
||
|
|
-------------------------------
|
||
|
|
|
||
|
|
.. autoclass:: torchreid.models.resnet.ResNet
|
||
|
|
.. autoclass:: torchreid.models.resnext.ResNeXt
|
||
|
|
.. autoclass:: torchreid.models.senet.SENet
|
||
|
|
.. autoclass:: torchreid.models.densenet.DenseNet
|
||
|
|
.. autoclass:: torchreid.models.inceptionresnetv2.InceptionResNetV2
|
||
|
|
.. autoclass:: torchreid.models.inceptionv4.InceptionV4
|
||
|
|
.. autoclass:: torchreid.models.xception.Xception
|
||
|
|
|
||
|
|
|
||
|
|
Lightweight Models
|
||
|
|
------------------
|
||
|
|
|
||
|
|
.. autoclass:: torchreid.models.nasnet.NASNetAMobile
|
||
|
|
.. autoclass:: torchreid.models.mobilenetv2.MobileNetV2
|
||
|
|
.. autoclass:: torchreid.models.shufflenet.ShuffleNet
|
||
|
|
.. autoclass:: torchreid.models.squeezenet.SqueezeNet
|
||
|
|
|
||
|
|
|
||
|
|
ReID-specific Models
|
||
|
|
--------------------
|
||
|
|
|
||
|
|
.. autoclass:: torchreid.models.mudeep.MuDeep
|
||
|
|
.. autoclass:: torchreid.models.resnetmid.ResNetMid
|
||
|
|
.. autoclass:: torchreid.models.hacnn.HACNN
|
||
|
|
.. autoclass:: torchreid.models.pcb.PCB
|
||
|
|
.. autoclass:: torchreid.models.mlfn.MLFN
|