{"cells":[{"metadata":{},"cell_type":"markdown","source":"# Summary\n\nThis notebook merges images from 2019 and 2020 cassava plant disease classification competitions and zips them into a single archive for the further use. The `train.csv` contains image IDs and labels of all images."},{"metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"cell_type":"code","source":"import pandas as pd\nimport numpy as np\nimport os","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"# 2019 DATA"},{"metadata":{"_kg_hide-output":true,"trusted":true},"cell_type":"code","source":"!unzip -q ../input/cassava-disease/train.zip -d train_images","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"train_folder = 'train_images/train/'\nlabels = os.listdir(train_folder)\nlabels","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"df = pd.DataFrame(columns = ['image_id', 'label'])","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"for label in labels:\n    for image in os.listdir(os.path.join(train_folder, label)):\n        df = pd.concat([df,\n                        pd.DataFrame({'image_id': [image],\n                                      'label':    [label]})],\n                        axis = 0)\n        \ndf.reset_index(drop = True)\ndf.head()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"mapping = {'healthy': 4,\n           'cmd':     3,\n           'cgm':     2,\n           'cbsd':    1,\n           'cbb':     0}\n\ndf['label'] = df['label'].map(mapping) \ndf['label'].value_counts()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"df['source'] = 2019","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"!rm -rf train_images/\n!unzip -q -j ../input/cassava-disease/train.zip -d train_images","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"# 2020 DATA"},{"metadata":{"trusted":true},"cell_type":"code","source":"!cp ../input/cassava-leaf-disease-classification/train_images/* train_images","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"!zip -q -r images.zip *\n!rm -rf train_images/","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"df2020 = pd.read_csv('/kaggle/input/cassava-leaf-disease-classification/train.csv')\ndf2020['source'] = 2020\ndf2020.head()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"df = pd.concat([df, df2020], axis = 0)\ndf['source'].value_counts()","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"df.to_csv('train.csv', index = False)","execution_count":null,"outputs":[]}],"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":4,"nbformat_minor":4}