{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"# This Python 3 environment comes with many helpful analytics libraries installed\n# It is defined by the kaggle/python Docker image: https://github.com/kaggle/docker-python\n# For example, here's several helpful packages to load\n\nimport numpy as np # linear algebra\nimport pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)\n\n# Input data files are available in the read-only \"../input/\" directory\n# For example, running this (by clicking run or pressing Shift+Enter) will list all files under the input directory\n\nimport os\nfor dirname, _, filenames in os.walk('/kaggle/input'):\n    for filename in filenames:\n        print(os.path.join(dirname, filename))\n\n# You can write up to 20GB to the current directory (/kaggle/working/) that gets preserved as output when you create a version using \"Save & Run All\" \n# You can also write temporary files to /kaggle/temp/, but they won't be saved outside of the current session","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2023-07-19T04:15:38.532357Z","iopub.execute_input":"2023-07-19T04:15:38.532802Z","iopub.status.idle":"2023-07-19T04:15:38.541721Z","shell.execute_reply.started":"2023-07-19T04:15:38.532758Z","shell.execute_reply":"2023-07-19T04:15:38.540674Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 必要なライブラリをインポート","metadata":{}},{"cell_type":"code","source":"from tensorflow import keras\nfrom keras.applications.vgg16 import VGG16\nfrom keras.preprocessing.image import ImageDataGenerator\nfrom keras.models import Sequential\nfrom keras.layers import Dense, Dropout, Flatten\nfrom keras.callbacks import ModelCheckpoint\nfrom matplotlib import pyplot as plt\nimport shutil\nimport os\nimport glob","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:38.543792Z","iopub.execute_input":"2023-07-19T04:15:38.544405Z","iopub.status.idle":"2023-07-19T04:15:38.556209Z","shell.execute_reply.started":"2023-07-19T04:15:38.544371Z","shell.execute_reply":"2023-07-19T04:15:38.555263Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Zipファイルを解凍","metadata":{}},{"cell_type":"code","source":"shutil.unpack_archive('/kaggle/input/dogs-vs-cats-redux-kernels-edition/train.zip', '/kaggle/working')\nshutil.unpack_archive('/kaggle/input/dogs-vs-cats-redux-kernels-edition/test.zip', '/kaggle/working')\n\n\nshutil.copyfile('/kaggle/input/dogs-vs-cats-redux-kernels-edition/sample_submission.csv','/kaggle/working/sample_submission.csv')\n\nfor dirname, _, filenames in os.walk('/kaggle/input'):\n    for filename in filenames:\n        print(os.path.join(dirname, filename))\n        \nfiles = glob.glob('/kaggle/working/*')\nfor file in files:\n    print(file)","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:38.557455Z","iopub.execute_input":"2023-07-19T04:15:38.557890Z","iopub.status.idle":"2023-07-19T04:15:48.494664Z","shell.execute_reply.started":"2023-07-19T04:15:38.557856Z","shell.execute_reply":"2023-07-19T04:15:48.493574Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":" # 画像サイズとディレクトリの設定","metadata":{}},{"cell_type":"code","source":"img_width, img_height = 150, 150\ntrain_dir = '/kaggle/working/train'\ntest_dir = '/kaggle/working/test'","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.497364Z","iopub.execute_input":"2023-07-19T04:15:48.497813Z","iopub.status.idle":"2023-07-19T04:15:48.502785Z","shell.execute_reply.started":"2023-07-19T04:15:48.497777Z","shell.execute_reply":"2023-07-19T04:15:48.501436Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# エポック数の設定","metadata":{}},{"cell_type":"code","source":"epoch = 20","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.504234Z","iopub.execute_input":"2023-07-19T04:15:48.504896Z","iopub.status.idle":"2023-07-19T04:15:48.513907Z","shell.execute_reply.started":"2023-07-19T04:15:48.504864Z","shell.execute_reply":"2023-07-19T04:15:48.512886Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 分類クラス名の設定","metadata":{}},{"cell_type":"code","source":"classes = ['dog','cat']\nnb_classes = len(classes)","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.515505Z","iopub.execute_input":"2023-07-19T04:15:48.516349Z","iopub.status.idle":"2023-07-19T04:15:48.524838Z","shell.execute_reply.started":"2023-07-19T04:15:48.516312Z","shell.execute_reply":"2023-07-19T04:15:48.523918Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# VGG16モデルのロード","metadata":{}},{"cell_type":"code","source":"vgg_model = VGG16(include_top = False, weights = 'imagenet', input_shape = (img_height, img_width, 3))","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.525846Z","iopub.execute_input":"2023-07-19T04:15:48.526283Z","iopub.status.idle":"2023-07-19T04:15:48.838729Z","shell.execute_reply.started":"2023-07-19T04:15:48.526251Z","shell.execute_reply":"2023-07-19T04:15:48.837672Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# VGG16モデルの下に全結合層を追加","metadata":{}},{"cell_type":"code","source":"model = Sequential()\nmodel.add(vgg_model)\nmodel.add(Flatten())\nmodel.add(Dense(256, activation='relu'))\nmodel.add(Dropout(0.5))\nmodel.add(Dense(nb_classes, activation='softmax'))","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.840088Z","iopub.execute_input":"2023-07-19T04:15:48.840439Z","iopub.status.idle":"2023-07-19T04:15:48.935652Z","shell.execute_reply.started":"2023-07-19T04:15:48.840407Z","shell.execute_reply":"2023-07-19T04:15:48.934680Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# VGG16モデルの上位15層のパラメータを凍結(Fine tuning)","metadata":{}},{"cell_type":"code","source":"for layer in vgg_model.layers[:15]:\n layer.trainable = False","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.937326Z","iopub.execute_input":"2023-07-19T04:15:48.937714Z","iopub.status.idle":"2023-07-19T04:15:48.944938Z","shell.execute_reply.started":"2023-07-19T04:15:48.937680Z","shell.execute_reply":"2023-07-19T04:15:48.943970Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# モデルの詳細を表示","metadata":{}},{"cell_type":"code","source":"model.summary()","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.949513Z","iopub.execute_input":"2023-07-19T04:15:48.950219Z","iopub.status.idle":"2023-07-19T04:15:48.974891Z","shell.execute_reply.started":"2023-07-19T04:15:48.950194Z","shell.execute_reply":"2023-07-19T04:15:48.974188Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 損失関数は交差エントロピー、最適化関数は確率的勾配法\n","metadata":{}},{"cell_type":"code","source":"model.compile(loss = 'categorical_crossentropy', optimizer = 'sgd', metrics = ['accuracy'])","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.975893Z","iopub.execute_input":"2023-07-19T04:15:48.976298Z","iopub.status.idle":"2023-07-19T04:15:48.992308Z","shell.execute_reply.started":"2023-07-19T04:15:48.976266Z","shell.execute_reply":"2023-07-19T04:15:48.991408Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 犬・猫の画像を，それぞれ保存するフォルダのパスを定義","metadata":{}},{"cell_type":"code","source":"dog_dir = '/kaggle/working/train/dog'\ncat_dir = '/kaggle/working/train/cat'\n\nos.makedirs(dog_dir, exist_ok = True)\nos.makedirs(cat_dir, exist_ok = True)\n","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:48.993607Z","iopub.execute_input":"2023-07-19T04:15:48.994571Z","iopub.status.idle":"2023-07-19T04:15:48.999669Z","shell.execute_reply.started":"2023-07-19T04:15:48.994538Z","shell.execute_reply":"2023-07-19T04:15:48.998463Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 犬・猫の画像をそれぞれのフォルダに移動  \n# 訓練用フォルダ内の全ファイルパスを取得","metadata":{}},{"cell_type":"code","source":"files = glob.glob('/kaggle/working/train/*.jpg')\nfor file in files:\n    file_name = os.path.basename(file)\n    if 'cat' in file:\n        shutil.move(file,'/kaggle/working/train/cat/' + file_name)\n    else:\n        shutil.move(file,'/kaggle/working/train/dog/' + file_name)","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:49.001049Z","iopub.execute_input":"2023-07-19T04:15:49.002372Z","iopub.status.idle":"2023-07-19T04:15:51.620660Z","shell.execute_reply.started":"2023-07-19T04:15:49.002346Z","shell.execute_reply":"2023-07-19T04:15:51.619684Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 学習データのデータ拡張を設定","metadata":{}},{"cell_type":"code","source":"train_datagen = ImageDataGenerator(rescale = 1.0 / 255, rotation_range = 90, shear_range = 0.2, zoom_range = 0.2, horizontal_flip = True,\n                vertical_flip = True, fill_mode = 'reflect', validation_split = 0.2)","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:51.622064Z","iopub.execute_input":"2023-07-19T04:15:51.622421Z","iopub.status.idle":"2023-07-19T04:15:51.628456Z","shell.execute_reply.started":"2023-07-19T04:15:51.622388Z","shell.execute_reply":"2023-07-19T04:15:51.627262Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 評価データのデータ拡張を設定","metadata":{}},{"cell_type":"code","source":"test_datagen = ImageDataGenerator(rescale = 1.0 / 255)","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:51.630022Z","iopub.execute_input":"2023-07-19T04:15:51.630721Z","iopub.status.idle":"2023-07-19T04:15:51.640538Z","shell.execute_reply.started":"2023-07-19T04:15:51.630686Z","shell.execute_reply":"2023-07-19T04:15:51.639486Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 学習データのジェネレータを生成","metadata":{}},{"cell_type":"code","source":"train_generator = train_datagen.flow_from_directory(train_dir, target_size = (img_height, img_width), classes = classes, batch_size = 32, class_mode = 'categorical', subset = \"training\")","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:51.642398Z","iopub.execute_input":"2023-07-19T04:15:51.642751Z","iopub.status.idle":"2023-07-19T04:15:52.238298Z","shell.execute_reply.started":"2023-07-19T04:15:51.642684Z","shell.execute_reply":"2023-07-19T04:15:52.237349Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 評価データのジェネレータを生成","metadata":{}},{"cell_type":"code","source":"test_generator = train_datagen.flow_from_directory(train_dir, target_size = (img_height, img_width), classes = classes, batch_size = 32, class_mode = 'categorical', subset = \"validation\")","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:52.239688Z","iopub.execute_input":"2023-07-19T04:15:52.240037Z","iopub.status.idle":"2023-07-19T04:15:52.462128Z","shell.execute_reply.started":"2023-07-19T04:15:52.240003Z","shell.execute_reply":"2023-07-19T04:15:52.461148Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# コールバック関数(モデル保存)の設定","metadata":{}},{"cell_type":"code","source":"mc_cb = ModelCheckpoint(filepath = 'finetuning.h5', monitor = 'vol_loss', verbose = 1, save_best_only = True)","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:52.465495Z","iopub.execute_input":"2023-07-19T04:15:52.465802Z","iopub.status.idle":"2023-07-19T04:15:52.470737Z","shell.execute_reply.started":"2023-07-19T04:15:52.465775Z","shell.execute_reply":"2023-07-19T04:15:52.469827Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# ジェネレータを用いたモデル学習","metadata":{}},{"cell_type":"code","source":"history = model.fit_generator(train_generator, epochs=epoch, validation_data = test_generator, callbacks = [mc_cb])\n\nmodel.save('finetuning.h5')","metadata":{"execution":{"iopub.status.busy":"2023-07-19T04:15:52.472555Z","iopub.execute_input":"2023-07-19T04:15:52.473382Z","iopub.status.idle":"2023-07-19T05:12:55.842029Z","shell.execute_reply.started":"2023-07-19T04:15:52.473349Z","shell.execute_reply":"2023-07-19T05:12:55.840643Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 予想誤差のグラフ化","metadata":{}},{"cell_type":"code","source":"plt.plot(range(len(history.history['loss'])), history.history['loss'], marker='o', color = 'black', label='loss')\nplt.plot(range(len(history.history['val_loss'])), history.history['val_loss'], marker='v', linestyle='--', color='red', label='val_loss')\nplt.xlabel('epoch')\nplt.ylabel('loss')\nplt.legend(loc='best')\nplt.show()","metadata":{"execution":{"iopub.status.busy":"2023-07-19T05:12:55.844734Z","iopub.execute_input":"2023-07-19T05:12:55.845114Z","iopub.status.idle":"2023-07-19T05:12:56.123535Z","shell.execute_reply.started":"2023-07-19T05:12:55.845072Z","shell.execute_reply":"2023-07-19T05:12:56.122577Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 予想精度のグラフ化","metadata":{}},{"cell_type":"code","source":"plt.plot(range(len(history.history['accuracy'])), history.history['accuracy'], marker='o', color = 'black', label='acc')\nplt.plot(range(len(history.history['val_accuracy'])), history.history['val_accuracy'], marker='v', linestyle='--', color = 'red', label='val_acc')\nplt.xlabel('epoch')\nplt.ylabel('accuracy')\nplt.legend(loc='best')\nplt.show()","metadata":{"execution":{"iopub.status.busy":"2023-07-19T05:12:56.125124Z","iopub.execute_input":"2023-07-19T05:12:56.125472Z","iopub.status.idle":"2023-07-19T05:12:56.409498Z","shell.execute_reply.started":"2023-07-19T05:12:56.125438Z","shell.execute_reply":"2023-07-19T05:12:56.408597Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 分類予測に必要なライブラリをインポート","metadata":{}},{"cell_type":"code","source":"import sys\nimport glob\nimport numpy as np\nfrom keras.preprocessing import image\nfrom keras.models import load_model","metadata":{"execution":{"iopub.status.busy":"2023-07-19T05:12:56.411043Z","iopub.execute_input":"2023-07-19T05:12:56.411413Z","iopub.status.idle":"2023-07-19T05:12:56.416210Z","shell.execute_reply.started":"2023-07-19T05:12:56.411379Z","shell.execute_reply":"2023-07-19T05:12:56.415227Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 画像サイズの設定","metadata":{}},{"cell_type":"code","source":"img_height, img_width = 150, 150","metadata":{"execution":{"iopub.status.busy":"2023-07-19T05:12:56.417780Z","iopub.execute_input":"2023-07-19T05:12:56.418474Z","iopub.status.idle":"2023-07-19T05:12:56.429254Z","shell.execute_reply.started":"2023-07-19T05:12:56.418441Z","shell.execute_reply":"2023-07-19T05:12:56.428290Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# 分類クラス名の設定","metadata":{}},{"cell_type":"code","source":"classes = ['dog','cat']\nnb_classes = len(classes)","metadata":{"execution":{"iopub.status.busy":"2023-07-19T05:12:56.430394Z","iopub.execute_input":"2023-07-19T05:12:56.431487Z","iopub.status.idle":"2023-07-19T05:12:56.439837Z","shell.execute_reply.started":"2023-07-19T05:12:56.431455Z","shell.execute_reply":"2023-07-19T05:12:56.438613Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# ファインチューニングしたCNNモデルのロード","metadata":{}},{"cell_type":"code","source":"model = load_model('finetuning.h5')","metadata":{"execution":{"iopub.status.busy":"2023-07-19T05:12:56.441421Z","iopub.execute_input":"2023-07-19T05:12:56.441859Z","iopub.status.idle":"2023-07-19T05:12:56.859806Z","shell.execute_reply.started":"2023-07-19T05:12:56.441827Z","shell.execute_reply":"2023-07-19T05:12:56.858704Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# テストフォルダー内の全画像ファイル名を取得","metadata":{}},{"cell_type":"code","source":"filename = glob.glob('/content/drive/MyDrive/data/test/*/*.jpg')","metadata":{"execution":{"iopub.status.busy":"2023-07-19T05:12:56.861455Z","iopub.execute_input":"2023-07-19T05:12:56.862059Z","iopub.status.idle":"2023-07-19T05:12:56.867189Z","shell.execute_reply.started":"2023-07-19T05:12:56.862023Z","shell.execute_reply":"2023-07-19T05:12:56.866175Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# テストフォルダ内の全画像を分類予測し各クラスの分類確率値を出力","metadata":{}},{"cell_type":"code","source":"for i in range(len(filename)):\n print('input:', filename[i])\n\n# 入力画像のロード、4次元テンソルへ変換 \n img = image.load_img(filename[i], target_size = (img_height, img_width))\n x = image.img_to_array(img)\n x = np.expand_dims(x, axis = 0)\n\n# 入力データへの正規化\n x = x / 255.0\n\n# 分類クラスを予測\n pred = model.predict(x)[0]\n\n#予測結果を予測確率が上位10件分、クラス名と予測確率を出力\n top_n = 10\n top_indices = pred.argsort()[-top_n:][::-1]\n result = [(classes[i], pred[i]) for i in top_indices]\n for x in result:\n  print(x)","metadata":{"execution":{"iopub.status.busy":"2023-07-19T05:12:56.868443Z","iopub.execute_input":"2023-07-19T05:12:56.869495Z","iopub.status.idle":"2023-07-19T05:12:56.879934Z","shell.execute_reply.started":"2023-07-19T05:12:56.869459Z","shell.execute_reply":"2023-07-19T05:12:56.878790Z"},"trusted":true},"execution_count":null,"outputs":[]}]}