2019-03-24 07:09:39 +08:00
|
|
|
.. _torchreid_models:
|
|
|
|
|
|
2019-03-22 08:14:41 +08:00
|
|
|
torchreid.models
|
|
|
|
|
=================
|
|
|
|
|
|
|
|
|
|
Interface
|
|
|
|
|
---------
|
|
|
|
|
|
|
|
|
|
.. automodule:: torchreid.models.__init__
|
2019-03-24 07:09:39 +08:00
|
|
|
:members:
|
2019-03-22 08:14:41 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
ImageNet Classification Models
|
|
|
|
|
-------------------------------
|
|
|
|
|
|
|
|
|
|
.. autoclass:: torchreid.models.resnet.ResNet
|
|
|
|
|
.. 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
|
2019-05-24 23:30:11 +08:00
|
|
|
.. autoclass:: torchreid.models.shufflenetv2.ShuffleNetV2
|
2019-03-22 08:14:41 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
ReID-specific Models
|
|
|
|
|
--------------------
|
|
|
|
|
|
|
|
|
|
.. autoclass:: torchreid.models.mudeep.MuDeep
|
|
|
|
|
.. autoclass:: torchreid.models.resnetmid.ResNetMid
|
|
|
|
|
.. autoclass:: torchreid.models.hacnn.HACNN
|
|
|
|
|
.. autoclass:: torchreid.models.pcb.PCB
|
2019-07-03 20:45:40 +08:00
|
|
|
.. autoclass:: torchreid.models.mlfn.MLFN
|
|
|
|
|
.. autoclass:: torchreid.models.osnet.OSNet
|
