41 lines
1.1 KiB
ReStructuredText
41 lines
1.1 KiB
ReStructuredText
.. _torchreid_models:
|
|
|
|
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 |