diff --git a/embedding-calculator/src/services/facescan/plugins/agegender/__init__.py b/embedding-calculator/src/services/facescan/plugins/agegender/__init__.py index dccb4a7368..77306a2dca 100644 --- a/embedding-calculator/src/services/facescan/plugins/agegender/__init__.py +++ b/embedding-calculator/src/services/facescan/plugins/agegender/__init__.py @@ -12,4 +12,4 @@ # or implied. See the License for the specific language governing # permissions and limitations under the License. -requirements = ('tensorflow~=2.1.4', 'tf-slim~=1.1.0') +requirements = ('tensorflow~=2.11.0', 'tf-slim~=1.1.0') diff --git a/embedding-calculator/src/services/facescan/plugins/dependencies.py b/embedding-calculator/src/services/facescan/plugins/dependencies.py index f5ff371f1b..c238a82c94 100644 --- a/embedding-calculator/src/services/facescan/plugins/dependencies.py +++ b/embedding-calculator/src/services/facescan/plugins/dependencies.py @@ -18,7 +18,7 @@ from src.services.utils.pyutils import get_env -def get_tensorflow(version='2.1.4') -> Tuple[str, ...]: +def get_tensorflow(version='2.11.0') -> Tuple[str, ...]: libs = [f'tensorflow=={version}'] cuda_version = get_env('CUDA', '').replace('.', '') if ENV.GPU_IDX > -1 and cuda_version: diff --git a/embedding-calculator/src/services/facescan/plugins/facenet/__init__.py b/embedding-calculator/src/services/facescan/plugins/facenet/__init__.py index 0171a996e6..0c0ab221e7 100644 --- a/embedding-calculator/src/services/facescan/plugins/facenet/__init__.py +++ b/embedding-calculator/src/services/facescan/plugins/facenet/__init__.py @@ -14,4 +14,4 @@ from src.services.facescan.plugins.dependencies import get_tensorflow -requirements = ('protobuf~=3.20.1',) + get_tensorflow() # + ('mtcnn~=0.1.0',) +requirements = ('protobuf~=3.19.6',) + get_tensorflow() # + ('mtcnn~=0.1.0',)