From e42a6d272adfbba22e24c96e9b41be5aef42f3e9 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Fri, 4 Nov 2016 14:00:19 +0100
Subject: [PATCH] VGG 16
---
bob/learn/tensorflow/layers/__init__.py | 1 +
bob/learn/tensorflow/network/__init__.py | 1 +
2 files changed, 2 insertions(+)
diff --git a/bob/learn/tensorflow/layers/__init__.py b/bob/learn/tensorflow/layers/__init__.py
index 7c2ce7f1..8cbd8692 100644
--- a/bob/learn/tensorflow/layers/__init__.py
+++ b/bob/learn/tensorflow/layers/__init__.py
@@ -7,6 +7,7 @@ from .Layer import Layer
from .Conv2D import Conv2D
from .FullyConnected import FullyConnected
from .MaxPooling import MaxPooling
+from .AveragePooling import AveragePooling
from .Dropout import Dropout
from .InputLayer import InputLayer
diff --git a/bob/learn/tensorflow/network/__init__.py b/bob/learn/tensorflow/network/__init__.py
index 0accdd4b..e583ea98 100644
--- a/bob/learn/tensorflow/network/__init__.py
+++ b/bob/learn/tensorflow/network/__init__.py
@@ -12,6 +12,7 @@ from .MLP import MLP
from .FaceNet import FaceNet
from .FaceNetSimple import FaceNetSimple
from .VGG16 import VGG16
+from .VGG16_mod import VGG16_mod
# gets sphinx autodoc done right - don't remove it
__all__ = [_ for _ in dir() if not _.startswith('_')]
--
GitLab