diff --git a/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile b/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile
index a0e7964560..edbdf7370a 100644
--- a/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile
+++ b/tests/metrics/machine_learning/resnet50_fp32_dockerfile/Dockerfile
@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 # Usage: FROM [image name]
-FROM ubuntu:20.04
+FROM intel/intel-optimized-tensorflow:2.9.1
 
 ENV DEBIAN_FRONTEND=noninteractive
 
diff --git a/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile b/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile
index a033475ee5..6e0c1e6119 100644
--- a/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile
+++ b/tests/metrics/machine_learning/resnet50_int8_dockerfile/Dockerfile
@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 # Usage: FROM [image name]
-FROM ubuntu:20.04
+FROM intel/intel-optimized-tensorflow:2.9.1
 
 ENV DEBIAN_FRONTEND=noninteractive