Extract graph and layer types and names to build Graph representation python (#133)

Author: Semyon1104

app/AlexNet/CMakeLists.txt

@@ -0,0 +1,41 @@
+import tensorflow as tf
+from tensorflow.keras.models import load_model
+
+# Путь к вашей модели .h5
+MODEL_PATH = 'cnn_cat_dog.h5'
+
+# Загрузка модели
+model = load_model(MODEL_PATH)
+
+# Получение графа модели
+graph = tf.compat.v1.get_default_graph()
+
+# Получение весов модели
+weights = model.get_weights()
+
+# Вывод структуры модели
+print("Model summary:")
+model.summary()
+OUTPUT_FILE = 'model_summary.txt'
+with open(OUTPUT_FILE, 'w') as f:
+    # Вывод структуры модели в файл
+    f.write("Model summary:\n")
+    model.summary(print_fn=lambda x: f.write(x + '\n'))
+
+    # Вывод параметров каждого слоя в файл
+    f.write("\nParameters of each layer:\n")
+    for layer in model.layers:
+        f.write("Layer Name: {}\n".format(layer.name))
+        f.write("Layer Trainable: {}\n".format(layer.trainable))
+        f.write("Layer Parameters: {}\n".format(layer.count_params()))
+        if hasattr(layer, 'get_weights'):
+            weights = layer.get_weights()
+            if weights:
+                f.write("Layer Weights:\n")
+                for i, w in enumerate(weights):
+                    f.write("Weight {}:\n".format(i + 1))
+                    f.write(str(w) + '\n')
+        f.write("=" * 50 + '\n')
+
+print("Model summary and layer parameters have been saved to", OUTPUT_FILE)
+

app/AlexNet/alexnet.cpp

@@ -1,46 +1 @@
 add_subdirectory(ReaderImage)
-if(UNIX)
-    find_path(tensorflow_INCLUDE_DIR
-            NAMES
-            tensorflow/core
-            tensorflow/c
-            HINTS
-            /usr/local/include/tensorflow
-            /usr/include/tensorflow)
-    find_library(tensorflow_LIBRARY NAMES tensorflow
-            HINTS
-            /usr/lib
-            /usr/local/lib)
-    include(FindPackageHandleStandardArgs)
-    find_package_handle_standard_args(tensorflow DEFAULT_MSG tensorflow_INCLUDE_DIR tensorflow_LIBRARY)
-    if(TENSORFLOW_FOUND)
-        set(TensorFlow_LIBRARIES ${tensorflow_LIBRARY})
-        set(TensorFlow_INCLUDE_DIRS ${tensorflow_INCLUDE_DIR})
-        SET(TENSORFLOW_FOUND "TRUE" CACHE INTERNAL "TENSORFLOW_FOUND")
-
-    else()
-        message(WARNING "TensorFlow not found. Skipping building of alexnet target.")
-        SET(TENSORFLOW_FOUND "FALSE" CACHE INTERNAL "TENSORFLOW_FOUND")
-        return()
-    endif()
-endif()
-
-if(WIN32)
-    set(TensorFlow_INCLUDE_DIRS "${CMAKE_SOURCE_DIR}/3rdparty/tensorflow/include")
-    set(TensorFlow_LIBRARIES "${CMAKE_SOURCE_DIR}/3rdparty/tensorflow/lib/tensorflow.lib")
-    if(EXISTS "${TensorFlow_INCLUDE_DIRS}" AND EXISTS "${TensorFlow_LIBRARIES}")
-         include(FindPackageHandleStandardArgs)
-         find_package_handle_standard_args(tensorflow DEFAULT_MSG TensorFlow_INCLUDE_DIRS TensorFlow_LIBRARIES)
-         SET(TENSORFLOW_FOUND "TRUE" CACHE INTERNAL "TENSORFLOW_FOUND")
-    else()
-         SET(TENSORFLOW_FOUND "FALSE" CACHE INTERNAL "TENSORFLOW_FOUND")
-         return()
-    endif()
-endif()
-
-if (TENSORFLOW_FOUND)
-    add_subdirectory(AlexNet)
-else()
-    message(WARNING "TensorFlow not found. Skipping building of alexnet target.")
-endif()
-

app/AlexNet/alexnet.hpp

@@ -1,9 +1,8 @@
-if(TENSORFLOW_FOUND)
-    file(GLOB_RECURSE TEST_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}/*.cpp)
-else()
-    file(GLOB_RECURSE TEST_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}/*.cpp)
-    list(REMOVE_ITEM TEST_SRC_FILES "${CMAKE_CURRENT_SOURCE_DIR}/model_read/model_read.cpp")
-endif()
+
+
+file(GLOB_RECURSE TEST_SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}/*.cpp)
+list(REMOVE_ITEM TEST_SRC_FILES "${CMAKE_CURRENT_SOURCE_DIR}/model_read/model_read.cpp")
+
 
 if (NOT WIN32)
     find_package(OpenMP REQUIRED)
@@ -16,14 +15,9 @@ endif()
 target_link_libraries(run_test PUBLIC perf_lib layers_lib)
 target_link_libraries(run_test PUBLIC gtest_main)
 target_link_libraries(run_test PUBLIC ReadLib)
-if(TENSORFLOW_FOUND)
-    target_link_libraries(run_test PUBLIC AlexNetLib)
-endif()
+
 
 target_include_directories(run_test PRIVATE "${CMAKE_SOURCE_DIR}/app/ReaderImage")
-if(TENSORFLOW_FOUND)
-    target_include_directories(run_test PRIVATE "${CMAKE_SOURCE_DIR}/app/AlexNet")
-endif()
 
 if (WIN32)
     add_custom_command(TARGET run_test POST_BUILD
@@ -69,4 +63,4 @@ file(DOWNLOAD
 )
 
 target_compile_definitions(run_test PUBLIC TESTS_BINARY_PATH="${CMAKE_CURRENT_BINARY_DIR}")
-add_definitions(-DMODEL_PATH="${CMAKE_CURRENT_BINARY_DIR}/squeeze.tflite")
+