{ "cells": [ { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [], "source": [ "from PIL import Image\n", "import os" ] }, { "cell_type": "code", "execution_count": 19, "metadata": {}, "outputs": [], "source": [ "dataset_path = \"/home/jbuisine/Documents/Thesis/Development/thesis-data/SIN3D_references/SIN3D_HD_all_center/\"\n", "output_path = \"tmp_images/SIN3D_references\"" ] }, { "cell_type": "code", "execution_count": 22, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "['p3d_bunny-fur-view0_part6', 'p3d_coffee-splash-view0_part6', 'p3d_ecosys-view0_part6', 'p3d_eponge-fractal-6-view0_part6', 'p3d_pavilion-night-view2_part6', 'p3d_arcsphere-view0_part6', 'p3d_bathroom-view0_part6', 'p3d_car2-view0_part6', 'p3d_caustic-view0_part6', 'p3d_classroom-view0_part6', 'p3d_classroom-view1_part6', 'p3d_contemporary-bathroom-view0_part6', 'p3d_contemporary-bathroom-view1_part6', 'p3d_crown-view0_part6', 'p3d_dragon-view0_part6', 'p3d_dragon_250-view0_part6', 'p3d_eponge-fractal-6-view1_part6', 'p3d_ganesha-view0_part6', 'p3d_glass-of-water-view0_part6', 'p3d_glass-view0_part6', 'p3d_indirect-view0_part6', 'p3d_kitchen-view0_part6', 'p3d_kitchen-view1_part6', 'p3d_landscape-view3_part6', 'p3d_living-room-3-view0_part6', 'p3d_living-room-3-view1_part6', 'p3d_low_table_2spheric_view0_part6', 'p3d_pavilion-day-view0_part6', 'p3d_pavilion-day-view1_part6', 'p3d_pavilion-day-view2_part6', 'p3d_pavilion-night-view0_part6', 'p3d_pavilion-night-view1_part6', 'p3d_sanmiguel-view1_part6', 'p3d_sanmiguel-view2_part6', 'p3d_staircase2-view0_part6', 'p3d_staircase-view1_part6', 'p3d_vw-van-view0_part6', 'p3d_tt-view0_part6', 'p3d_living-room-view0_part6', 'p3d_living-room-view2_part6']\n" ] } ], "source": [ "thresholds_file_path = \"/home/jbuisine/Downloads/SVD-Entropy_SIN3D/thresholds_SVD-Entropy.csv\"\n", "\n", "scenes_kept = []\n", "with open(thresholds_file_path, 'r') as f:\n", " for line in f.readlines():\n", " scenes_kept.append(line.split(';')[0])\n", " \n", "print(scenes_kept)" ] }, { "cell_type": "code", "execution_count": 24, "metadata": {}, "outputs": [], "source": [ "if not os.path.exists(output_path):\n", " os.makedirs(output_path)" ] }, { "cell_type": "code", "execution_count": 25, "metadata": {}, "outputs": [], "source": [ "references_images = sorted(os.listdir(dataset_path))" ] }, { "cell_type": "code", "execution_count": 48, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "p3d_arcsphere-view0_part6.png\n", "p3d_bathroom-view0_part6.png\n", "p3d_bedroom-view0_part6.png\n", "p3d_bedroom-view1_part6.png\n", "p3d_bidir-view0_part6.png\n", "p3d_bmw-m6-view0_part6.png\n", "p3d_bunny-fur-view0_part6.png\n", "p3d_car2-view0_part6.png\n", "p3d_caustic-view0_part6.png\n", "p3d_chopper-titan-view0_part6.png\n", "p3d_classroom-view0_part6.png\n", "p3d_classroom-view1_part6.png\n", "p3d_coffee-splash-view0_part6.png\n", "p3d_contemporary-bathroom-view0_part6.png\n", "p3d_contemporary-bathroom-view1_part6.png\n", "p3d_cornel-box-view0_part6.png\n", "p3d_crown-view0_part6.png\n", "p3d_dining-room-view0_part6.png\n", "p3d_dining-room-view1_part6.png\n", "p3d_dragon-view0_part6.png\n", "p3d_dragon_250-view0_part6.png\n", "p3d_echecs-view0_part6.png\n", "p3d_echecs-view1_part6.png\n", "p3d_echecs-view2_part6.png\n", "p3d_ecosys-view0_part6.png\n", "p3d_eponge-fractal-5-view0_part6.png\n", "p3d_eponge-fractal-5-view1_part6.png\n", "p3d_eponge-fractal-6-view0_part6.png\n", "p3d_eponge-fractal-6-view1_part6.png\n", "p3d_ganesha-view0_part6.png\n", "p3d_glass-of-water-view0_part6.png\n", "p3d_glass-view0_part6.png\n", "p3d_glasses-view0_part6.png\n", "p3d_indirect-view0_part6.png\n", "p3d_kitchen-view0_part6.png\n", "p3d_kitchen-view1_part6.png\n", "p3d_lamp-view0_part6.png\n", "p3d_landscape-view0_part6.png\n", "p3d_landscape-view1_part6.png\n", "p3d_landscape-view2_part6.png\n", "p3d_landscape-view3_part6.png\n", "p3d_landscape-view4_part6.png\n", "p3d_living-room-2-view0_part6.png\n", "p3d_living-room-2-view1_part6.png\n", "p3d_living-room-3-view0_part6.png\n", "p3d_living-room-3-view1_part6.png\n", "p3d_living-room-view0_part6.png\n", "p3d_living-room-view1_part6.png\n", "p3d_living-room-view2_part6.png\n", "p3d_low_table_2spheric_view0_part6.png\n", "p3d_low_table_coridor_view0_part6.png\n", "p3d_low_table_upper_view0_part6.png\n", "p3d_low_table_window-view0_part6.png\n", "p3d_pavilion-day-view0_part6.png\n", "p3d_pavilion-day-view1_part6.png\n", "p3d_pavilion-day-view2_part6.png\n", "p3d_pavilion-night-view0_part6.png\n", "p3d_pavilion-night-view1_part6.png\n", "p3d_pavilion-night-view2_part6.png\n", "p3d_rubix-view0_part6.png\n", "p3d_sanmiguel-view0_part6.png\n", "p3d_sanmiguel-view1_part6.png\n", "p3d_sanmiguel-view2_part6.png\n", "p3d_sanmiguel-view3_part6.png\n", "p3d_spaceship-view0_part6.png\n", "p3d_sportscar-view0_part6.png\n", "p3d_staircase-view0_part6.png\n", "p3d_staircase-view1_part6.png\n", "p3d_staircase2-view0_part6.png\n", "p3d_staircase2-view1_part6.png\n", "p3d_tt-view0_part6.png\n", "p3d_tungsten_veach-mis-view0_part6.png\n", "p3d_veach-ajar-view0_part6.png\n", "p3d_veach-ajar-view1_part6.png\n", "p3d_villa-daylight-view0_part6.png\n", "p3d_villa-daylight-view1_part6.png\n", "p3d_villa-daylight-view2_part6.png\n", "p3d_vw-van-view0_part6.png\n", "p3d_whiteroom-daytime-view0_part6.png\n", "p3d_whiteroom-night-view0_part6.png\n" ] } ], "source": [ "for ref in references_images:\n", " print(ref)\n", " img_path = os.path.join(dataset_path, ref)\n", " pil_img = Image.open(img_path)\n", " pil_img = pil_img.resize((160, 160))\n", " \n", " if ref.replace('.png', '') in scenes_kept:\n", " output_path_img_path = os.path.join(output_path, ref)\n", " pil_img.save(output_path_img_path)" ] }, { "cell_type": "code", "execution_count": 46, "metadata": {}, "outputs": [], "source": [ "message = \"\t\\\\begin{{subfigure}}[t]{{0.15\\\\textwidth}} \\n \\\n", "\t\t\\t\\includegraphics[width=\\\\textwidth]{{{0}}} \\n \\\n", "\t\\end{{subfigure}}\" \\" ] }, { "cell_type": "code", "execution_count": 54, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_arcsphere-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_bathroom-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_bunny-fur-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_car2-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_caustic-view0_part6.png} \n", " \t\\end{subfigure}\n", "\n", "\t\\vspace{2mm}\n", "\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_classroom-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_classroom-view1_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_coffee-splash-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_contemporary-bathroom-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_contemporary-bathroom-view1_part6.png} \n", " \t\\end{subfigure}\n", "\n", "\t\\vspace{2mm}\n", "\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_crown-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_dragon-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_dragon_250-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_ecosys-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_eponge-fractal-6-view0_part6.png} \n", " \t\\end{subfigure}\n", "\n", "\t\\vspace{2mm}\n", "\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_eponge-fractal-6-view1_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_ganesha-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_glass-of-water-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_glass-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_indirect-view0_part6.png} \n", " \t\\end{subfigure}\n", "\n", "\t\\vspace{2mm}\n", "\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_kitchen-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_kitchen-view1_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_landscape-view3_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_living-room-3-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_living-room-3-view1_part6.png} \n", " \t\\end{subfigure}\n", "\n", "\t\\vspace{2mm}\n", "\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_living-room-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_living-room-view2_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_low_table_2spheric_view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_pavilion-day-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_pavilion-day-view1_part6.png} \n", " \t\\end{subfigure}\n", "\n", "\t\\vspace{2mm}\n", "\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_pavilion-day-view2_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_pavilion-night-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_pavilion-night-view1_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_pavilion-night-view2_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_sanmiguel-view1_part6.png} \n", " \t\\end{subfigure}\n", "\n", "\t\\vspace{2mm}\n", "\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_sanmiguel-view2_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_staircase-view1_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_staircase2-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_tt-view0_part6.png} \n", " \t\\end{subfigure}\n", "\t~\n", "\t\\begin{subfigure}[t]{0.15\\textwidth} \n", " \t\t\t\\includegraphics[width=\\textwidth]{images/2-Dataset/all_images/p3d_vw-van-view0_part6.png} \n", " \t\\end{subfigure}\n", "\n", "\t\\vspace{2mm}\n", "\n" ] } ], "source": [ "prefix_path = \"images/2-Dataset/all_images\"\n", "\n", "counter = 1\n", "for ref in references_images:\n", " \n", " if ref.replace('.png', '') in scenes_kept:\n", " output_message = message.format(os.path.join(prefix_path, ref))\n", " print(output_message)\n", " \n", " if counter % 5 == 0:\n", " print(\"\\n\\t\\\\vspace{2mm}\\n\")\n", " else:\n", " print(\"\\t~\")\n", " \n", " counter += 1" ] } ], "metadata": { "kernelspec": { "display_name": "Python 3.6.1 64-bit ('thesis-venv': venv)", "language": "python", "name": "python36164bitthesisvenvvenva88eb0ffd2f54d649db9e765b609711d" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.6.1" } }, "nbformat": 4, "nbformat_minor": 4 }