mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add ref

pull/17/head
KaiyangZhou 2018-05-12 22:49:34 +01:00
parent 21e21cdd9d
commit d8482a4185
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
models/DPN.py
View File

@ -16,8 +16,12 @@ many ideas from another PyTorch implementation https://github.com/oyam/pytorch-D
This implementation is compatible with the pretrained weights This implementation is compatible with the pretrained weights
from cypw's MXNet implementation. from cypw's MXNet implementation.
""" """
#__all__ = ['DPN', 'dpn68', 'dpn68b', 'dpn92', 'dpn98', 'dpn131', 'dpn107'] #__all__ = ['DPN', 'dpn68', 'dpn68b', 'dpn92', 'dpn98', 'dpn131', 'dpn107']
"""
Code imported from https://github.com/Cadene/pretrained-models.pytorch
"""
__all__ = ['DPN'] __all__ = ['DPN']
pretrained_settings = { pretrained_settings = {

4
models/InceptionResNetV2.py
View File

@ -7,6 +7,10 @@ import torch.utils.model_zoo as model_zoo
import os import os
import sys import sys
"""
Code imported from https://github.com/Cadene/pretrained-models.pytorch
"""
__all__ = ['InceptionResNetV2'] __all__ = ['InceptionResNetV2']
pretrained_settings = { pretrained_settings = {