###get_tensor_by_nameに関するエラーがどうしてもエラーが取れません。
githubにあるコードを動かしたいです。
コードが古いので、そのせいかもしれません。
python
1# tf.summary.FileWriter('log_dir', model.graph) 2# !tensorboard --logdir='log_dir/' 3 4y_pred = model.graph.get_tensor_by_name('model_1/activation_1/Softmax:0') 5y_logits = model.graph.get_tensor_by_name('model_1/logits/BiasAdd:0') 6input_img = model.graph.get_tensor_by_name('input_1:0')
エラー文
AttributeError Traceback (most recent call last)
<ipython-input-20-1bc04bf241b4> in <module>
----> 1 y_pred = model.graph.get_tensor_by_name('model/activation_1/Softmax:0')
2 y_logits = model.graph.get_tensor_by_name('model/logits/BiasAdd:0')
3 input_img = model.graph.get_tensor_by_name('input_1:0')
AttributeError: 'Model' object has no attribute 'graph'
補足
コメントアウトした部分で、TensorBoardを確認出来る様ですが、
こちらを実行すると'Model' object has no attribute 'graph'という同じエラーが出ます。
どうやらmodelに問題がありそうです。
試した事
検討違いかもしれませんが、
「get_tensor_by_name」で検索し、検索結果から
「get_tensor_by_name」の前にあるコードを真似て書き換えを行いましたが同じエラーが出ました。
環境
Anaconda、JupiterNotebook、Mac、python3.7
importしたもの
python-opencv 4.4.0.46
tensorflow 2.00
numpy 1.19.2
コードとファイル構造
直接は関係無いかと思いますが、一応載せました。
ファイル構造
killme_adversarial_example_with_TensorFlow-Keras-master
|—(下記の本コード).py
|—kill_me_images
| |—gen_adv
| |—kill_me_baby_dataset
|
|—models(作った)
コード
同じファイル(実行部分)のコードになります。
※上記のコードを含む
import tensorflow as tf import numpy as np import matplotlib import matplotlib.pyplot as plt %matplotlib inline import os import cv2 from PIL import Image config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True)) sess = tf.compat.v1.Session(config=config) tf.compat.v1.keras.backend.set_session(sess) root_dir = '/Users/ユーザー名/Desktop/killme_adversarial_example_with_TensorFlow-Keras-master/kill_me_images/kill_me_baby_datasets/' class_list = sorted(os.listdir(path=root_dir)) print("クラスリスト", class_list) def load_imgs(root_dir): print(class_list) num_class = len(class_list) img_paths = [] labels = [] images = [] for cl_name in class_list: img_names = os.listdir(os.path.join(root_dir, cl_name)) for img_name in img_names: img_paths.append(os.path.abspath(os.path.join(root_dir, cl_name, img_name))) hot_cl_name = get_class_one_hot(cl_name, class_list) labels.append(hot_cl_name) for img_path in img_paths: img = cv2.imread(img_path) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) images.append(img) images = np.array(images) return np.array(images), np.array(labels), class_list ############ ここから学習モデルを作成 ########### batch_size = 32 vgg16 = tf.keras.applications.VGG16(include_top=False, weights='imagenet', classes=7, input_shape=(128, 128, 3)) vgg16.trainable = False inputs = tf.keras.Input(shape=vgg16.output.shape[1:]) x = tf.keras.layers.Flatten()(inputs) x = tf.keras.layers.Dense(256, activation='relu')(x) x = tf.keras.layers.Dropout(0.5)(x) logits = tf.keras.layers.Dense(len(class_list), name='logits')(x) out = tf.keras.layers.Activation('softmax')(logits) top_model = tf.keras.Model(inputs=inputs, outputs=out) model = tf.keras.Model(inputs=vgg16.input, outputs=top_model(vgg16.output)) # 最後の畳み込み層の前までの重みの更新はしない。 # 15層目が最後の畳み込み層の開始位置 for layer in model.layers[:15]: layer.trainable = False # 学習方法を設定する model.compile(loss='categorical_crossentropy', optimizer=tf.keras.optimizers.Adam(), metrics=['accuracy']) tb_cb = tf.keras.callbacks.TensorBoard(log_dir="log_dir") ckps = [tb_cb] # 学習用データを用意する train_datagen = tf.keras.preprocessing.image.ImageDataGenerator(rescale=1./255, horizontal_flip=True) train_generator = train_datagen.flow_from_directory('kill_me_images/kill_me_baby_datasets/', classes=class_list, target_size=(128, 128), batch_size=batch_size, class_mode='categorical') # 学習開始! model.fit_generator(train_generator, steps_per_epoch=train_generator.samples//batch_size, epochs=100, callbacks=ckps) model.save("models/killme_vgg16.h5") sess = tf.keras.backend.get_session() saver = tf.train.Saver() saver.save(sess, "models/killme_vgg16.ckpt") # tf.summary.FileWriter('log_dir', model.graph) # !tensorboard --logdir='log_dir/' y_pred = model.graph.get_tensor_by_name('model_1/activation_1/Softmax:0') y_logits = model.graph.get_tensor_by_name('model_1/logits/BiasAdd:0') input_img = model.graph.get_tensor_by_name('input_1:0')