{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.7.12","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"none","dataSources":[{"sourceId":6799,"databundleVersionId":4225553,"sourceType":"competition"}],"dockerImageVersionId":30357,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import os\nimport shutil\nfrom tqdm import tqdm\nimport random\nimport numpy as np\n\ntrain_path = \"/kaggle/input/imagenet-object-localization-challenge/ILSVRC/Data/CLS-LOC/train\"","metadata":{"execution":{"iopub.status.busy":"2024-02-12T02:16:05.331634Z","iopub.execute_input":"2024-02-12T02:16:05.331985Z","iopub.status.idle":"2024-02-12T02:16:05.336685Z","shell.execute_reply.started":"2024-02-12T02:16:05.331958Z","shell.execute_reply":"2024-02-12T02:16:05.33598Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"ex_per_class = 10\noutput_folder = \"/kaggle/working/imagenet_1pct\"\n\n\nclasses = np.random.choice(os.listdir(train_path), 10, replace=False)\n\n\nfor folder in tqdm(classes):\n    os.makedirs(os.path.join(output_folder, folder))\n    folder_path = os.path.join(train_path, folder)\n    folder_files = os.listdir(folder_path)\n    \n    # Select random images\n    selected_files = folder_files # np.random.choice(folder_files, ex_per_class, replace=False)\n    \n    # Copy selected files to output folder\n    for filename in selected_files:\n        shutil.copy2(\n            os.path.join(folder_path, filename),\n            os.path.join(output_folder, folder),\n        )\n        \nshutil.make_archive(\"imagenet_subset\", 'zip', output_folder)","metadata":{"execution":{"iopub.status.busy":"2024-02-12T02:16:07.840714Z","iopub.execute_input":"2024-02-12T02:16:07.841101Z","iopub.status.idle":"2024-02-12T02:18:44.192187Z","shell.execute_reply.started":"2024-02-12T02:16:07.841072Z","shell.execute_reply":"2024-02-12T02:18:44.190469Z"},"trusted":true},"execution_count":null,"outputs":[]}]}