{"metadata":{"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"name":"python","version":"3.6.6","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"gpu","dataSources":[{"sourceId":3362,"databundleVersionId":31148,"sourceType":"competition"},{"sourceId":7174783,"sourceType":"datasetVersion","datasetId":4145935},{"sourceId":7175303,"sourceType":"datasetVersion","datasetId":4146306}],"dockerImageVersionId":25160,"isInternetEnabled":false,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"This Kernel for someone want to deep dive into image classification. I use CNN for classification model. If you found this Kernel helpful please up vote it. If you have some feedback and question don't forget to comment below. \n\nI have simplier model with \n* https://www.kaggle.com/uysimty/get-start-image-classification","metadata":{}},{"cell_type":"markdown","source":"# Import Library","metadata":{"_uuid":"fe76d1d1ded592430e7548feacfa38dc42f085d9"}},{"cell_type":"code","source":"import numpy as np\nimport pandas as pd \nfrom keras.preprocessing.image import ImageDataGenerator, load_img\nfrom keras.utils import to_categorical\nfrom sklearn.model_selection import train_test_split\nimport matplotlib.pyplot as plt\nimport random\nimport os\nprint(os.listdir(\"../input/dogs-vs-cats\"))\n!unzip -q ../input/dogs-vs-cats/train.zip\n!unzip -q ../input/dogs-vs-cats/test1.zip\nglobal data_test \ndata_test = \"mydata\" # 保证和你的数据名字一致，这里是mydata\n!cp -r ../input/{data_test} /kaggle/working/","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2023-12-26T03:59:20.305851Z","iopub.execute_input":"2023-12-26T03:59:20.306176Z","iopub.status.idle":"2023-12-26T03:59:32.655776Z","shell.execute_reply.started":"2023-12-26T03:59:20.306127Z","shell.execute_reply":"2023-12-26T03:59:32.654880Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Define Constants","metadata":{}},{"cell_type":"code","source":"FAST_RUN = False\nIMAGE_WIDTH=128\nIMAGE_HEIGHT=128\nIMAGE_SIZE=(IMAGE_WIDTH, IMAGE_HEIGHT)\nIMAGE_CHANNELS=3","metadata":{"execution":{"iopub.status.busy":"2023-12-26T03:59:32.657456Z","iopub.execute_input":"2023-12-26T03:59:32.657742Z","iopub.status.idle":"2023-12-26T03:59:32.663008Z","shell.execute_reply.started":"2023-12-26T03:59:32.657689Z","shell.execute_reply":"2023-12-26T03:59:32.662150Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Prepare Traning Data","metadata":{"_uuid":"7335a579cc0268fba5d34d6f7558f33c187eedb3"}},{"cell_type":"code","source":"filenames = os.listdir(\"./train/\")\ncategories = []\nfor filename in filenames:\n    category = filename.split('.')[0]\n    if category == 'dog':\n        categories.append(1)\n    else:\n        categories.append(0)\n\ndf = pd.DataFrame({\n    'filename': filenames,\n    'category': categories\n})","metadata":{"_cell_guid":"79c7e3d0-c299-4dcb-8224-4455121ee9b0","_uuid":"d629ff2d2480ee46fbb7e2d37f6b5fab8052498a","execution":{"iopub.status.busy":"2023-12-26T03:59:32.664344Z","iopub.execute_input":"2023-12-26T03:59:32.664589Z","iopub.status.idle":"2023-12-26T03:59:32.720319Z","shell.execute_reply.started":"2023-12-26T03:59:32.664538Z","shell.execute_reply":"2023-12-26T03:59:32.719466Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"df.head()","metadata":{"_uuid":"915bb9ba7063ab4d5c07c542419ae119003a5f98","execution":{"iopub.status.busy":"2023-12-26T03:59:32.721656Z","iopub.execute_input":"2023-12-26T03:59:32.721904Z","iopub.status.idle":"2023-12-26T03:59:32.734830Z","shell.execute_reply.started":"2023-12-26T03:59:32.721855Z","shell.execute_reply":"2023-12-26T03:59:32.734147Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"df.tail()","metadata":{"_uuid":"72bf69e817f67f5a2eaff8561217e22077248553","execution":{"iopub.status.busy":"2023-12-26T03:59:32.736281Z","iopub.execute_input":"2023-12-26T03:59:32.736546Z","iopub.status.idle":"2023-12-26T03:59:32.752971Z","shell.execute_reply.started":"2023-12-26T03:59:32.736502Z","shell.execute_reply":"2023-12-26T03:59:32.752031Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### See Total In count","metadata":{"_uuid":"a999484fc35b73373fafe2253ae9db7ff46fdb90"}},{"cell_type":"code","source":"df['category'].value_counts().plot.bar()","metadata":{"_uuid":"fa26f0bc7a6d835a24989790b20f3c6f32946f45","execution":{"iopub.status.busy":"2023-12-26T03:59:32.754186Z","iopub.execute_input":"2023-12-26T03:59:32.754547Z","iopub.status.idle":"2023-12-26T03:59:33.018371Z","shell.execute_reply.started":"2023-12-26T03:59:32.754490Z","shell.execute_reply":"2023-12-26T03:59:33.017317Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"From our data we have 12000 cats and 12000 dogs","metadata":{"_uuid":"3a08da58107777a1dd05c4a4bf5c484484923cac"}},{"cell_type":"markdown","source":"# See sample image","metadata":{"_uuid":"400a293df3c8499059d9175f3915187074efd971"}},{"cell_type":"code","source":"sample = random.choice(filenames)\nimage = load_img(\"./train/\"+sample)\nplt.imshow(image)","metadata":{"_uuid":"602b40f7353871cb161c60b5237f0da0096b2f47","execution":{"iopub.status.busy":"2023-12-26T03:59:33.020045Z","iopub.execute_input":"2023-12-26T03:59:33.020640Z","iopub.status.idle":"2023-12-26T03:59:33.274096Z","shell.execute_reply.started":"2023-12-26T03:59:33.020573Z","shell.execute_reply":"2023-12-26T03:59:33.273276Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Build Model\n\n<img src=\"https://i.imgur.com/ebkMGGu.jpg\" width=\"100%\"/>","metadata":{"_uuid":"b244e6b7715a04fc6df92dd6dfa3d35c473ca600"}},{"cell_type":"markdown","source":"* **Input Layer**: It represent input image data. It will reshape image into single diminsion array. Example your image is 64x64 = 4096, it will convert to (4096,1) array.\n* **Conv Layer**: This layer will extract features from image.\n* **Pooling Layer**: This layerreduce the spatial volume of input image after convolution.\n* **Fully Connected Layer**: It connect the network from a layer to another layer\n* **Output Layer**: It is the predicted values layer. ","metadata":{}},{"cell_type":"code","source":"from keras.models import Sequential\nfrom keras.layers import Conv2D, MaxPooling2D, Dropout, Flatten, Dense, Activation, BatchNormalization\n\nmodel = Sequential()\n\nmodel.add(Conv2D(32, (3, 3), activation='relu', input_shape=(IMAGE_WIDTH, IMAGE_HEIGHT, IMAGE_CHANNELS)))\nmodel.add(BatchNormalization())\nmodel.add(MaxPooling2D(pool_size=(2, 2)))\nmodel.add(Dropout(0.25))\n\nmodel.add(Conv2D(64, (3, 3), activation='relu'))\nmodel.add(BatchNormalization())\nmodel.add(MaxPooling2D(pool_size=(2, 2)))\nmodel.add(Dropout(0.25))\n\nmodel.add(Conv2D(128, (3, 3), activation='relu'))\nmodel.add(BatchNormalization())\nmodel.add(MaxPooling2D(pool_size=(2, 2)))\nmodel.add(Dropout(0.25))\n\nmodel.add(Flatten())\nmodel.add(Dense(512, activation='relu'))\nmodel.add(BatchNormalization())\nmodel.add(Dropout(0.5))\nmodel.add(Dense(2, activation='softmax')) # 2 because we have cat and dog classes\n\nmodel.compile(loss='categorical_crossentropy', optimizer='rmsprop', metrics=['accuracy'])\n\nmodel.summary()","metadata":{"_uuid":"8c9f833c1441b657c779844912d0b8028218d454","execution":{"iopub.status.busy":"2023-12-26T03:59:33.275239Z","iopub.execute_input":"2023-12-26T03:59:33.275498Z","iopub.status.idle":"2023-12-26T03:59:35.612808Z","shell.execute_reply.started":"2023-12-26T03:59:33.275456Z","shell.execute_reply":"2023-12-26T03:59:35.612094Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Callbacks","metadata":{"_uuid":"bd496f6c65888a969be3703135b0b03a8a1190c8"}},{"cell_type":"code","source":"from keras.callbacks import EarlyStopping, ReduceLROnPlateau","metadata":{"_uuid":"9aa032f0f6da539d23918890d2d131cc3aac8c7a","execution":{"iopub.status.busy":"2023-12-26T03:59:35.614113Z","iopub.execute_input":"2023-12-26T03:59:35.614358Z","iopub.status.idle":"2023-12-26T03:59:35.618500Z","shell.execute_reply.started":"2023-12-26T03:59:35.614316Z","shell.execute_reply":"2023-12-26T03:59:35.617288Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Early Stop**\n\nTo prevent over fitting we will stop the learning after 10 epochs and val_loss value not decreased","metadata":{"_uuid":"76c9ba4fb7f930c96b2c3e0d6b68ed9fa6a4227b"}},{"cell_type":"code","source":"earlystop = EarlyStopping(patience=10)","metadata":{"_uuid":"3421c5ec428da6c0d8cc1184179a9caff1e01d1c","execution":{"iopub.status.busy":"2023-12-26T03:59:35.619819Z","iopub.execute_input":"2023-12-26T03:59:35.620075Z","iopub.status.idle":"2023-12-26T03:59:35.627926Z","shell.execute_reply.started":"2023-12-26T03:59:35.620034Z","shell.execute_reply":"2023-12-26T03:59:35.627394Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Learning Rate Reduction**\n\nWe will reduce the learning rate when then accuracy not increase for 2 steps","metadata":{"_uuid":"51d3fe52e911286433cedf6e47332948a253361e"}},{"cell_type":"code","source":"learning_rate_reduction = ReduceLROnPlateau(monitor='val_acc', \n                                            patience=2, \n                                            verbose=1, \n                                            factor=0.5, \n                                            min_lr=0.00001)","metadata":{"_uuid":"8010a5661ad8924d2db24af0f3c00b1593b38901","execution":{"iopub.status.busy":"2023-12-26T03:59:35.629187Z","iopub.execute_input":"2023-12-26T03:59:35.629437Z","iopub.status.idle":"2023-12-26T03:59:35.638359Z","shell.execute_reply.started":"2023-12-26T03:59:35.629388Z","shell.execute_reply":"2023-12-26T03:59:35.637539Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"callbacks = [earlystop, learning_rate_reduction]","metadata":{"_uuid":"a79cc604199469789f183096d863f7248e5f6aab","execution":{"iopub.status.busy":"2023-12-26T03:59:35.639653Z","iopub.execute_input":"2023-12-26T03:59:35.640034Z","iopub.status.idle":"2023-12-26T03:59:35.648085Z","shell.execute_reply.started":"2023-12-26T03:59:35.639894Z","shell.execute_reply":"2023-12-26T03:59:35.647364Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Prepare data","metadata":{}},{"cell_type":"markdown","source":"Because we will use image genaretor `with class_mode=\"categorical\"`. We need to convert column category into string. Then imagenerator will convert it one-hot encoding which is good for our classification. \n\nSo we will convert 1 to dog and 0 to cat","metadata":{}},{"cell_type":"code","source":"df[\"category\"] = df[\"category\"].replace({0: 'cat', 1: 'dog'}) ","metadata":{"execution":{"iopub.status.busy":"2023-12-26T03:59:35.649475Z","iopub.execute_input":"2023-12-26T03:59:35.649760Z","iopub.status.idle":"2023-12-26T03:59:35.662417Z","shell.execute_reply.started":"2023-12-26T03:59:35.649707Z","shell.execute_reply":"2023-12-26T03:59:35.661669Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_df, validate_df = train_test_split(df, test_size=0.20, random_state=42)\ntrain_df = train_df.reset_index(drop=True)\nvalidate_df = validate_df.reset_index(drop=True)","metadata":{"_uuid":"4eeb7af8dcf02c4ef5ca744c8305c51a2f5cedef","execution":{"iopub.status.busy":"2023-12-26T03:59:35.663636Z","iopub.execute_input":"2023-12-26T03:59:35.663881Z","iopub.status.idle":"2023-12-26T03:59:35.673708Z","shell.execute_reply.started":"2023-12-26T03:59:35.663825Z","shell.execute_reply":"2023-12-26T03:59:35.672899Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_df['category'].value_counts().plot.bar()","metadata":{"_uuid":"b84836337441705eda9d2e655665ffa14d9feead","execution":{"iopub.status.busy":"2023-12-26T03:59:35.675081Z","iopub.execute_input":"2023-12-26T03:59:35.675383Z","iopub.status.idle":"2023-12-26T03:59:35.932402Z","shell.execute_reply.started":"2023-12-26T03:59:35.675331Z","shell.execute_reply":"2023-12-26T03:59:35.930804Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"validate_df['category'].value_counts().plot.bar()","metadata":{"_uuid":"19cf03f9a3c39532d6e2d06bd30be49a5afd9d57","execution":{"iopub.status.busy":"2023-12-26T03:59:35.936099Z","iopub.execute_input":"2023-12-26T03:59:35.936503Z","iopub.status.idle":"2023-12-26T03:59:36.211597Z","shell.execute_reply.started":"2023-12-26T03:59:35.936432Z","shell.execute_reply":"2023-12-26T03:59:36.210540Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"total_train = train_df.shape[0]\ntotal_validate = validate_df.shape[0]\nbatch_size=15","metadata":{"_uuid":"ae3dec0361f0443132d0309d3b883ee80070cf9f","execution":{"iopub.status.busy":"2023-12-26T03:59:36.213063Z","iopub.execute_input":"2023-12-26T03:59:36.213400Z","iopub.status.idle":"2023-12-26T03:59:36.220710Z","shell.execute_reply.started":"2023-12-26T03:59:36.213325Z","shell.execute_reply":"2023-12-26T03:59:36.219946Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Traning Generator","metadata":{"_uuid":"ff760be9104f7d9492467b8d9d3405011aa77d11"}},{"cell_type":"code","source":"train_datagen = ImageDataGenerator(\n    rotation_range=15,\n    rescale=1./255,\n    shear_range=0.1,\n    zoom_range=0.2,\n    horizontal_flip=True,\n    width_shift_range=0.1,\n    height_shift_range=0.1\n)\n\ntrain_generator = train_datagen.flow_from_dataframe(\n    train_df, \n    \"./train/\", \n    x_col='filename',\n    y_col='category',\n    target_size=IMAGE_SIZE,\n    class_mode='categorical',\n    batch_size=batch_size\n)","metadata":{"_uuid":"4d1c7818703a8a4bac5c036fdea45972aa9e5e9e","execution":{"iopub.status.busy":"2023-12-26T03:59:36.222122Z","iopub.execute_input":"2023-12-26T03:59:36.222715Z","iopub.status.idle":"2023-12-26T03:59:36.516927Z","shell.execute_reply.started":"2023-12-26T03:59:36.222663Z","shell.execute_reply":"2023-12-26T03:59:36.516051Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### Validation Generator","metadata":{"_uuid":"859c7b2857939c19fd2e3bb32839c9f7deb5aa3f"}},{"cell_type":"code","source":"validation_datagen = ImageDataGenerator(rescale=1./255)\nvalidation_generator = validation_datagen.flow_from_dataframe(\n    validate_df, \n    \"./train/\", \n    x_col='filename',\n    y_col='category',\n    target_size=IMAGE_SIZE,\n    class_mode='categorical',\n    batch_size=batch_size\n)","metadata":{"_uuid":"7925e16bcacc89f4484fb6fe47e54d6420af732e","execution":{"iopub.status.busy":"2023-12-26T03:59:36.518456Z","iopub.execute_input":"2023-12-26T03:59:36.518897Z","iopub.status.idle":"2023-12-26T03:59:36.591379Z","shell.execute_reply.started":"2023-12-26T03:59:36.518764Z","shell.execute_reply":"2023-12-26T03:59:36.590643Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# See how our generator work","metadata":{"_uuid":"6e17fc1f002fedd60febb78fee5e81770640b909"}},{"cell_type":"code","source":"example_df = train_df.sample(n=1).reset_index(drop=True)\nexample_generator = train_datagen.flow_from_dataframe(\n    example_df, \n    \"./train/\", \n    x_col='filename',\n    y_col='category',\n    target_size=IMAGE_SIZE,\n    class_mode='categorical'\n)","metadata":{"_uuid":"4252cce168ab65f88e44a8ebc2672607bc852af4","execution":{"iopub.status.busy":"2023-12-26T03:59:36.592840Z","iopub.execute_input":"2023-12-26T03:59:36.593138Z","iopub.status.idle":"2023-12-26T03:59:36.606976Z","shell.execute_reply.started":"2023-12-26T03:59:36.593083Z","shell.execute_reply":"2023-12-26T03:59:36.606313Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"plt.figure(figsize=(12, 12))\nfor i in range(0, 15):\n    plt.subplot(5, 3, i+1)\n    for X_batch, Y_batch in example_generator:\n        image = X_batch[0]\n        plt.imshow(image)\n        break\nplt.tight_layout()\nplt.show()","metadata":{"_uuid":"23d923dba747f8b47dc75569244cecc6f70df321","execution":{"iopub.status.busy":"2023-12-26T03:59:36.608378Z","iopub.execute_input":"2023-12-26T03:59:36.608704Z","iopub.status.idle":"2023-12-26T03:59:39.622116Z","shell.execute_reply.started":"2023-12-26T03:59:36.608645Z","shell.execute_reply":"2023-12-26T03:59:39.621359Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"Seem to be nice ","metadata":{"_uuid":"810ddf1373d9db470ed48da4f30ca5a6c1274435"}},{"cell_type":"markdown","source":"# Fit Model","metadata":{"_uuid":"5cd8df64e794ed17de326b613a9819e7da977a0e"}},{"cell_type":"code","source":"epochs=1 #if FAST_RUN else 50\nhistory = model.fit_generator(\n    train_generator, \n    epochs=epochs,\n    validation_data=validation_generator,\n    validation_steps=total_validate//batch_size,\n    steps_per_epoch=total_train//batch_size,\n    callbacks=callbacks\n)","metadata":{"_uuid":"0836a4cc8aa0abf603e0f96573c0c4ff383ad56b","execution":{"iopub.status.busy":"2023-12-26T03:59:39.623627Z","iopub.execute_input":"2023-12-26T03:59:39.623913Z","iopub.status.idle":"2023-12-26T04:02:53.215301Z","shell.execute_reply.started":"2023-12-26T03:59:39.623859Z","shell.execute_reply":"2023-12-26T04:02:53.214436Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Save Model","metadata":{"_uuid":"aa1fbc4081ae0de2993188b2bf658a0be5bc0687"}},{"cell_type":"code","source":"model.save_weights(\"model.h5\")","metadata":{"_uuid":"67575a4decdaf79a915d23151626b784ffa82642","execution":{"iopub.status.busy":"2023-12-26T04:02:53.217170Z","iopub.execute_input":"2023-12-26T04:02:53.217986Z","iopub.status.idle":"2023-12-26T04:02:53.434916Z","shell.execute_reply.started":"2023-12-26T04:02:53.217919Z","shell.execute_reply":"2023-12-26T04:02:53.434049Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Virtualize Training","metadata":{"_uuid":"1b76c0a9040bc0babf0a453e567e41e22f8a1e0e"}},{"cell_type":"code","source":"# fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(12, 12))\n# ax1.plot(history.history['loss'], color='b', label=\"Training loss\")\n# ax1.plot(history.history['val_loss'], color='r', label=\"validation loss\")\n# ax1.set_xticks(np.arange(1, epochs, 1))\n# ax1.set_yticks(np.arange(0, 1, 0.1))\n\n# ax2.plot(history.history['acc'], color='b', label=\"Training accuracy\")\n# ax2.plot(history.history['val_acc'], color='r',label=\"Validation accuracy\")\n# ax2.set_xticks(np.arange(1, epochs, 1))\n\n# legend = plt.legend(loc='best', shadow=True)\n# plt.tight_layout()\n# plt.show()","metadata":{"_uuid":"79055f2dc3e2abb47bea758e0464c86ca42ab431","execution":{"iopub.status.busy":"2023-12-26T04:02:53.436149Z","iopub.execute_input":"2023-12-26T04:02:53.436430Z","iopub.status.idle":"2023-12-26T04:02:53.439924Z","shell.execute_reply.started":"2023-12-26T04:02:53.436374Z","shell.execute_reply":"2023-12-26T04:02:53.439275Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Prepare Testing Data","metadata":{"_uuid":"764dc66e4b2bc558f3a0f90b80bb802f5b3d45a8"}},{"cell_type":"code","source":"test_filenames = os.listdir(\"./test1\")\ntest_df = pd.DataFrame({\n    'filename': test_filenames\n})\nnb_samples = test_df.shape[0]","metadata":{"_uuid":"c35e70d3e1e4834dbbf840fa0ea08c049bfcd915","execution":{"iopub.status.busy":"2023-12-26T04:02:53.441012Z","iopub.execute_input":"2023-12-26T04:02:53.441265Z","iopub.status.idle":"2023-12-26T04:02:53.463508Z","shell.execute_reply.started":"2023-12-26T04:02:53.441197Z","shell.execute_reply":"2023-12-26T04:02:53.462881Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Create Testing Generator","metadata":{"_uuid":"291bc3996dce8d05e174b27d64f03996d3e8038e"}},{"cell_type":"code","source":"test_gen = ImageDataGenerator(rescale=1./255)\ntest_generator = test_gen.flow_from_dataframe(\n    test_df, \n    \"./test1/\", \n    x_col='filename',\n    y_col=None,\n    class_mode=None,\n    target_size=IMAGE_SIZE,\n    batch_size=batch_size,\n    shuffle=False\n)","metadata":{"_uuid":"52249ec1c35fb1be3adef386be245de3794e55aa","execution":{"iopub.status.busy":"2023-12-26T04:02:53.464738Z","iopub.execute_input":"2023-12-26T04:02:53.464967Z","iopub.status.idle":"2023-12-26T04:02:53.602621Z","shell.execute_reply.started":"2023-12-26T04:02:53.464922Z","shell.execute_reply":"2023-12-26T04:02:53.601877Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Predict","metadata":{"_uuid":"2fa580afca2931ec5ce374e732d8c1789d03f2ed"}},{"cell_type":"code","source":"predict = model.predict_generator(test_generator, steps=np.ceil(nb_samples/batch_size))","metadata":{"_uuid":"4782eb23fa7d003f0e2415d995894017edb2d896","execution":{"iopub.status.busy":"2023-12-26T04:02:53.603983Z","iopub.execute_input":"2023-12-26T04:02:53.604254Z","iopub.status.idle":"2023-12-26T04:03:46.289254Z","shell.execute_reply.started":"2023-12-26T04:02:53.604174Z","shell.execute_reply":"2023-12-26T04:03:46.288426Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"For categoral classication the prediction will come with probability of each category. So we will pick the category that have the highest probability with numpy average max","metadata":{}},{"cell_type":"code","source":"test_df['category'] = np.argmax(predict, axis=-1)","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:03:46.290346Z","iopub.execute_input":"2023-12-26T04:03:46.290593Z","iopub.status.idle":"2023-12-26T04:03:46.296082Z","shell.execute_reply.started":"2023-12-26T04:03:46.290525Z","shell.execute_reply":"2023-12-26T04:03:46.295248Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"We will convert the predict category back into our generator classes by using `train_generator.class_indices`. It is the classes that image generator map while converting data into computer vision","metadata":{}},{"cell_type":"code","source":"label_map = dict((v,k) for k,v in train_generator.class_indices.items())\ntest_df['category'] = test_df['category'].replace(label_map)","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:03:46.297557Z","iopub.execute_input":"2023-12-26T04:03:46.297826Z","iopub.status.idle":"2023-12-26T04:03:46.311118Z","shell.execute_reply.started":"2023-12-26T04:03:46.297778Z","shell.execute_reply":"2023-12-26T04:03:46.310328Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"From our prepare data part. We map data with `{1: 'dog', 0: 'cat'}`. Now we will map the result back to dog is 1 and cat is 0","metadata":{}},{"cell_type":"code","source":"test_df['category'] = test_df['category'].replace({ 'dog': 1, 'cat': 0 })","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:03:46.312289Z","iopub.execute_input":"2023-12-26T04:03:46.312563Z","iopub.status.idle":"2023-12-26T04:03:46.332460Z","shell.execute_reply.started":"2023-12-26T04:03:46.312515Z","shell.execute_reply":"2023-12-26T04:03:46.331566Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### Visualize Result","metadata":{"_uuid":"b00add65fe765529e637c3a9904d710bb7eff1d8"}},{"cell_type":"code","source":"test_df['category'].value_counts().plot.bar()","metadata":{"_uuid":"d0bf6dd5ff344092fa0121f70bdd60fa5a40e29c","execution":{"iopub.status.busy":"2023-12-26T04:03:46.333498Z","iopub.execute_input":"2023-12-26T04:03:46.333784Z","iopub.status.idle":"2023-12-26T04:03:46.492610Z","shell.execute_reply.started":"2023-12-26T04:03:46.333731Z","shell.execute_reply":"2023-12-26T04:03:46.491692Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### See predicted result with images","metadata":{"_uuid":"ce72a83f80d6e012b12b82c8ee3365d671a3b307"}},{"cell_type":"code","source":"sample_test = test_df.head(18)\nsample_test.head()\nplt.figure(figsize=(12, 24))\nfor index, row in sample_test.iterrows():\n    filename = row['filename']\n    category = row['category']\n    img = load_img('test1/'+filename, target_size=IMAGE_SIZE)\n    plt.subplot(6, 3, index+1)\n    plt.imshow(img)\n    plt.xlabel(filename + '(' + \"{}\".format(category) + ')' )\nplt.tight_layout()\nplt.show()","metadata":{"_uuid":"98b41dc83075e6297137fb45bf703c313dd4ae28","execution":{"iopub.status.busy":"2023-12-26T04:03:46.494523Z","iopub.execute_input":"2023-12-26T04:03:46.495234Z","iopub.status.idle":"2023-12-26T04:03:51.105753Z","shell.execute_reply.started":"2023-12-26T04:03:46.495152Z","shell.execute_reply":"2023-12-26T04:03:51.104926Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"# Test your own dataset","metadata":{"_uuid":"d1ca25943e73aa20a37f9fb8670ee430caeaaf1f"}},{"cell_type":"markdown","source":"## Prepare your own data","metadata":{}},{"cell_type":"code","source":"my_files = f\"./{data_test}\"\nmy_filenames = os.listdir(my_files)\nmy_df = pd.DataFrame({\n    'filename': my_filenames\n})\nmy_samples = my_df.shape[0]","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:05:11.769592Z","iopub.execute_input":"2023-12-26T04:05:11.769882Z","iopub.status.idle":"2023-12-26T04:05:11.775719Z","shell.execute_reply.started":"2023-12-26T04:05:11.769836Z","shell.execute_reply":"2023-12-26T04:05:11.774637Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"## Creating my test generator","metadata":{}},{"cell_type":"code","source":"my_gen = ImageDataGenerator(rescale=1./255)\nmy_generator = my_gen.flow_from_dataframe(\n    my_df, \n    my_files, \n    x_col='filename',\n    y_col=None,\n    class_mode=None,\n    target_size=IMAGE_SIZE,\n    batch_size=batch_size,\n    shuffle=False\n)","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:05:21.278813Z","iopub.execute_input":"2023-12-26T04:05:21.279101Z","iopub.status.idle":"2023-12-26T04:05:21.289662Z","shell.execute_reply.started":"2023-12-26T04:05:21.279057Z","shell.execute_reply":"2023-12-26T04:05:21.288799Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"## Predict","metadata":{}},{"cell_type":"code","source":"my_predict = model.predict_generator(my_generator, steps=np.ceil(my_samples/batch_size))","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:05:24.781889Z","iopub.execute_input":"2023-12-26T04:05:24.782224Z","iopub.status.idle":"2023-12-26T04:05:24.813171Z","shell.execute_reply.started":"2023-12-26T04:05:24.782151Z","shell.execute_reply":"2023-12-26T04:05:24.812074Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"my_df['category'] = np.argmax(my_predict, axis=-1)","metadata":{"_uuid":"cce9f3e2ffff0693d79d84590ed71fbbca7c3c7c","execution":{"iopub.status.busy":"2023-12-26T04:05:29.445558Z","iopub.execute_input":"2023-12-26T04:05:29.446019Z","iopub.status.idle":"2023-12-26T04:05:29.450182Z","shell.execute_reply.started":"2023-12-26T04:05:29.445795Z","shell.execute_reply":"2023-12-26T04:05:29.449484Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"my_df['category'] = my_df['category'].replace(label_map)","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:05:33.491836Z","iopub.execute_input":"2023-12-26T04:05:33.492119Z","iopub.status.idle":"2023-12-26T04:05:33.498157Z","shell.execute_reply.started":"2023-12-26T04:05:33.492074Z","shell.execute_reply":"2023-12-26T04:05:33.497376Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"my_df['category'] = my_df['category'].replace({ 'dog': 1, 'cat': 0 })","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:05:34.056314Z","iopub.execute_input":"2023-12-26T04:05:34.056845Z","iopub.status.idle":"2023-12-26T04:05:34.065041Z","shell.execute_reply.started":"2023-12-26T04:05:34.056768Z","shell.execute_reply":"2023-12-26T04:05:34.064253Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"## Visualize Result","metadata":{}},{"cell_type":"code","source":"my_df['category'].value_counts().plot.bar()","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:05:36.078101Z","iopub.execute_input":"2023-12-26T04:05:36.078395Z","iopub.status.idle":"2023-12-26T04:05:36.309809Z","shell.execute_reply.started":"2023-12-26T04:05:36.078350Z","shell.execute_reply":"2023-12-26T04:05:36.308597Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"my_sample_test = my_df.head(6)\nmy_sample_test.head()\nplt.figure(figsize=(12, 9))\nfor index, row in my_sample_test.iterrows():\n    filename = row['filename']\n    category = row['category']\n    img = load_img(my_files+'/'+filename, target_size=IMAGE_SIZE)\n    plt.subplot(2, 3, index+1)\n    plt.imshow(img)\n    plt.xlabel(filename + '(' + \"{}\".format(category) + ')' )\nplt.tight_layout()\nplt.show()","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:06:24.685802Z","iopub.execute_input":"2023-12-26T04:06:24.686102Z","iopub.status.idle":"2023-12-26T04:06:26.097899Z","shell.execute_reply.started":"2023-12-26T04:06:24.686057Z","shell.execute_reply":"2023-12-26T04:06:26.096912Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"print(\"恭喜你成功完成实践！\")","metadata":{"execution":{"iopub.status.busy":"2023-12-26T04:09:31.170515Z","iopub.execute_input":"2023-12-26T04:09:31.170808Z","iopub.status.idle":"2023-12-26T04:09:31.175072Z","shell.execute_reply.started":"2023-12-26T04:09:31.170764Z","shell.execute_reply":"2023-12-26T04:09:31.174086Z"},"trusted":true},"execution_count":null,"outputs":[]}]}