add new ref to osnet.py

2019-10-22 22:46:30 +01:00 · 2019-10-22 22:46:30 +01:00 · 894de59bff
parent 1a5db72550
commit 894de59bff
1 changed files with 2 additions and 0 deletions
--- a/torchreid/models/osnet.py
+++ b/torchreid/models/osnet.py
@ -210,6 +210,8 @@ class OSNet(nn.Module):
    
    Reference:
        - Zhou et al. Omni-Scale Feature Learning for Person Re-Identification. ICCV, 2019.
+        - Zhou et al. Learning Generalisable Omni-Scale Representations
+          for Person Re-Identification. arXiv preprint, 2019.
    """

    def __init__(self, num_classes, blocks, layers, channels, feature_dim=512, loss='softmax', IN=False, **kwargs):