github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 19:45:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

tf2

Browse Source
This commit is contained in:
serengil 2020-12-13 20:10:41 +03:00
parent ada31621ac
commit 767b2028b5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deepface/basemodels/VGGFace.py
View File

@ -3,8 +3,9 @@ from pathlib import Path
import gdown
import tensorflow as tf
tf_version = int(tf.__version__.split(".")[0])
if int(tf.__version__.split(".")[0]) == 1:
if tf_version == 1:
from keras.models import Model, Sequential
from keras.layers import Input, Convolution2D, ZeroPadding2D, MaxPooling2D, Flatten, Dense, Dropout, Activation
else: