github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +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 gdown
import tensorflow as tf 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.models import Model, Sequential
from keras.layers import Input, Convolution2D, ZeroPadding2D, MaxPooling2D, Flatten, Dense, Dropout, Activation from keras.layers import Input, Convolution2D, ZeroPadding2D, MaxPooling2D, Flatten, Dense, Dropout, Activation
else: else: