From 86421e899c87976d8559795979415e3fae2bd7ed Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 31 Dec 2020 21:00:01 +0800 Subject: Implement some parts of RGB-GaitPart wrapper 1. Triplet loss function and weight init function haven't been implement yet 2. Tuplize features returned by auto-encoder for later unpack 3. Correct comment error in auto-encoder 4. Swap batch_size dim and time dim in HPM and PartNet in case of redundant transpose 5. Find backbone problems in HPM and disable it temporarily 6. Make feature structure by HPM consistent to that by PartNet 7. Fix average pooling dimension issue and incorrect view change in HP --- models/__init__.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'models/__init__.py') diff --git a/models/__init__.py b/models/__init__.py index 3b4d86e..51c86af 100644 --- a/models/__init__.py +++ b/models/__init__.py @@ -1 +1,4 @@ from .model import Model +from .auto_encoder import AutoEncoder +from .hpm import HorizontalPyramidMatching +from .part_net import PartNet -- cgit v1.2.3