deep-person-reid/torchreid/models
Ramakrishnan Sundareswaran 32acbbe4ad
Update mudeep.py to support testing
Current model fails in the case of triplet loss during testing since y,x returns a tuple where engine.py features expects a torch tensor instead to run features.cpu()
2021-04-08 11:31:27 -05:00
..
__init__.py linting and formatting code 2019-12-01 02:35:44 +00:00
densenet.py linting and formatting code 2019-12-02 14:34:59 +00:00
hacnn.py linting code 2020-05-05 15:58:00 +01:00
inceptionresnetv2.py linting code 2020-05-05 15:58:00 +01:00
inceptionv4.py linting and formatting code 2019-12-02 14:34:59 +00:00
mlfn.py linting code 2020-05-05 15:58:00 +01:00
mobilenetv2.py linting code 2020-05-05 15:58:00 +01:00
mudeep.py Update mudeep.py to support testing 2021-04-08 11:31:27 -05:00
nasnet.py linting code 2020-05-05 15:58:00 +01:00
osnet.py update reference 2021-03-26 11:45:06 +08:00
osnet_ain.py update reference 2021-03-26 11:45:06 +08:00
pcb.py linting and formatting code 2019-12-02 14:34:59 +00:00
resnet.py linting and formatting code 2019-12-02 14:34:59 +00:00
resnet_ibn_a.py linting and formatting code 2019-12-02 14:34:59 +00:00
resnet_ibn_b.py linting and formatting code 2019-12-02 14:34:59 +00:00
resnetmid.py linting and formatting code 2019-12-02 14:34:59 +00:00
senet.py linting and formatting code 2019-12-02 14:34:59 +00:00
shufflenet.py linting code 2020-05-05 15:58:00 +01:00
shufflenetv2.py linting and formatting code 2019-12-02 14:34:59 +00:00
squeezenet.py linting code 2020-05-05 15:58:00 +01:00
xception.py linting and formatting code 2019-12-02 14:34:59 +00:00