Commit 3afdb8f2 authored by Oldřich Pecák's avatar Oldřich Pecák
Browse files

progress on plotting stuff

parent f23c2d93

python/helper_bits.py

+1 −0
Original line number Diff line number Diff line 
import csv

import json

import os

import numpy as np



def loadconfig(filename):

python/plot_basic_tests.py

0 → 100644
+45 −0
Original line number Diff line number Diff line 
import matplotlib.pyplot as plt

import matplotlib as mpl

import os

from helper_bits import loaddata, loadconfig





data_list = ("sim_data/basic_tests/hole_sizes.json",

             "sim_data/basic_tests/uniform_grid.json",

             "sim_data/basic_tests/spacing_grid.json",

             "sim_data/basic_tests/spacing_lines.json",

             "sim_data/basic_tests/hole_sizes_sensor_shift.json",

             "sim_data/basic_tests/uniform_grid_sensor_shift.json",

             "sim_data/basic_tests/spacing_grid_sensor_shift.json",

             "sim_data/basic_tests/spacing_lines_sensor_shift.json"

             )

data = []



for item in data_list:

    config_abs = os.path.abspath(item)

    config = loadconfig(config_abs)

    data.append(loaddata(config, os.path.dirname(config_abs)))





fig, axs = plt.subplots(nrows=2, ncols=4, sharex=True, sharey=True)



colors = mpl.colormaps["viridis"]



axs[0, 0].imshow(data[0], interpolation='none', cmap=colors)

axs[0, 0].set_title("Hole sizes", fontsize='x-large')

axs[0, 0].set_ylabel("Target shift", fontsize='x-large')

axs[0, 1].imshow(data[1], interpolation='none', cmap=colors)

axs[0, 1].set_title("Uniform grid", fontsize='x-large')

axs[0, 2].imshow(data[2], interpolation='none', cmap=colors)

axs[0, 2].set_title("Spacing grid", fontsize='x-large')

axs[0, 3].imshow(data[3], interpolation='none', cmap=colors)

axs[0, 3].set_title("Spacing lines", fontsize='x-large')

axs[1, 0].imshow(data[4], interpolation='none', cmap=colors)

axs[1, 0].set_ylabel("Sensor shift", fontsize='x-large')

axs[1, 1].imshow(data[5], interpolation='none', cmap=colors)

axs[1, 2].imshow(data[6], interpolation='none', cmap=colors)

axs[1, 3].imshow(data[7], interpolation='none', cmap=colors)





fig.tight_layout()

plt.show()

python/plot_noise_processing.py

0 → 100644
+77 −0
Original line number Diff line number Diff line 
import matplotlib.pyplot as plt

import matplotlib as mpl

import numpy as np

import os

from functools import partial

from scipy import signal

from skimage import restoration

from helper_bits import loaddata, loadconfig





data_list = ("sim_data/noise_tests/uniform_grid_noise_high.json",

             "sim_data/noise_tests/hole_sizes_noise.json",

             "sim_data/noise_tests/spacing_grid_noise.json",

             "sim_data/noise_tests/spacing_lines_noise.json"

             )

data = []



for item in data_list:

    config_abs = os.path.abspath(item)

    config = loadconfig(config_abs)

    data.append(loaddata(config, os.path.dirname(config_abs)))



plt.figure(figsize=(20, 30))



gs = mpl.gridspec.GridSpec(8, 4, width_ratios=[1, 1, 1, 1],

         wspace=0.05, hspace=0.05, top=0.95, bottom=0.05, left=0.17, right=0.845)



colors = mpl.colormaps["viridis"]

f_size = "x-large"



ax=plt.subplot(gs[0, 0])

ax.imshow(data[0], interpolation='none', cmap=colors)

ax.set_title("Uniform grid", fontsize=f_size)

ax.set_ylabel("Original", fontsize=f_size)

ax.set_xticklabels([])

ax.set_yticklabels([])

ax=plt.subplot(gs[0, 1])

ax.imshow(data[1], interpolation='none', cmap=colors)

ax.set_title("Hole sizes", fontsize=f_size)

ax.set_xticklabels([])

ax.set_yticklabels([])

ax=plt.subplot(gs[0, 2])

ax.imshow(data[2], interpolation='none', cmap=colors)

ax.set_title("Spacing grid", fontsize=f_size)

ax.set_xticklabels([])

ax.set_yticklabels([])

ax=plt.subplot(gs[0, 3])

ax.imshow(data[3], interpolation='none', cmap=colors)

ax.set_title("Spacing lines", fontsize=f_size)

ax.set_xticklabels([])

ax.set_yticklabels([])



methods = [

    ("Median", partial(signal.medfilt2d, kernel_size=3)),

    ("Wiener", partial(signal.wiener, mysize=[3,3])),

    ("Bilateral", partial(restoration.denoise_bilateral)),

    ("Non-Local Means", partial(restoration.denoise_nl_means, patch_size=4)),

    ("Total Variation\n(Bregman)", partial(restoration.denoise_tv_bregman)),

    ("Total Variation\n(Chambolle)", partial(restoration.denoise_tv_chambolle)),

    ("Wavelet", partial(restoration.denoise_wavelet))

]



row_indx = 1

for name, func in methods:

    for index, item in enumerate(data):

        normalized = item/np.amax(item)

        filtered = func(normalized)

        ax = plt.subplot(gs[row_indx, index])

        if (index == 0):

            ax.set_ylabel(name, fontsize=f_size)

        ax.imshow(filtered, interpolation='none', cmap=colors)

        ax.set_xticklabels([])

        ax.set_yticklabels([])

    row_indx += 1



#plt.show()

plt.savefig("noise_processing.pdf", dpi=800, bbox_inches='tight')

python/plot_noise_tests.py

0 → 100644
+39 −0
Original line number Diff line number Diff line 
import matplotlib.pyplot as plt

import matplotlib as mpl

import os

from helper_bits import loaddata, loadconfig





data_list = ("sim_data/noise_tests/uniform_grid_noise_low.json",

             "sim_data/noise_tests/uniform_grid_noise_medium.json",

             "sim_data/noise_tests/uniform_grid_noise_high.json",

             "sim_data/noise_tests/hole_sizes_noise.json",

             "sim_data/noise_tests/spacing_grid_noise.json",

             "sim_data/noise_tests/spacing_lines_noise.json"

             )

data = []



for item in data_list:

    config_abs = os.path.abspath(item)

    config = loadconfig(config_abs)

    data.append(loaddata(config, os.path.dirname(config_abs)))



fig, axs = plt.subplots(nrows=2, ncols=3, sharex=True, sharey=True)



colors = mpl.colormaps["viridis"]



axs[0, 0].imshow(data[0], interpolation='none', cmap=colors)

axs[0, 0].set_title("Uniform grid 'low'", fontsize='x-large')

axs[0, 1].imshow(data[1], interpolation='none', cmap=colors)

axs[0, 1].set_title("Uniform grid 'medium'", fontsize='x-large')

axs[0, 2].imshow(data[2], interpolation='none', cmap=colors)

axs[0, 2].set_title("Uniform grid 'high'", fontsize='x-large')

axs[1, 0].imshow(data[3], interpolation='none', cmap=colors)

axs[1, 0].set_title("Hole sizes 'high'", fontsize='x-large')

axs[1, 1].imshow(data[4], interpolation='none', cmap=colors)

axs[1, 1].set_title("Spacing grid 'high'", fontsize='x-large')

axs[1, 2].imshow(data[5], interpolation='none', cmap=colors)

axs[1, 2].set_title("Spacing lines 'high'", fontsize='x-large')



fig.tight_layout()

plt.show()

python/plot_position_tests.py

0 → 100644
+47 −0
Original line number Diff line number Diff line 
import matplotlib.pyplot as plt

import matplotlib as mpl

import os

from helper_bits import loaddata, loadconfig





data_list = ("sim_data/position_tests/hole_sizes_near.json",

             "sim_data/position_tests/spacing_grid_near.json",

             "sim_data/position_tests/spacing_lines_near.json",

             "sim_data/position_tests/hole_sizes_mid.json",

             "sim_data/position_tests/spacing_grid_mid.json",

             "sim_data/position_tests/spacing_lines_mid.json",

             "sim_data/position_tests/hole_sizes_far.json",

             "sim_data/position_tests/spacing_grid_far.json",

             "sim_data/position_tests/spacing_lines_far.json"

             )

data = []



for item in data_list:

    config_abs = os.path.abspath(item)

    config = loadconfig(config_abs)

    data.append(loaddata(config, os.path.dirname(config_abs)))





fig, axs = plt.subplots(nrows=3, ncols=3, sharex=True, sharey=True)



colors = mpl.colormaps["viridis"]



axs[0, 0].imshow(data[0], interpolation='none', cmap=colors)

axs[0, 0].set_title("Hole sizes", fontsize='x-large')

axs[0, 0].set_ylabel("5 mm", fontsize='x-large')

axs[0, 1].imshow(data[1], interpolation='none', cmap=colors)

axs[0, 1].set_title("Spacing grid", fontsize='x-large')

axs[0, 2].imshow(data[2], interpolation='none', cmap=colors)

axs[0, 2].set_title("Spacing lines", fontsize='x-large')

axs[1, 0].imshow(data[3], interpolation='none', cmap=colors)

axs[1, 0].set_ylabel("25 mm", fontsize='x-large')

axs[1, 1].imshow(data[4], interpolation='none', cmap=colors)

axs[1, 2].imshow(data[5], interpolation='none', cmap=colors)

axs[2, 0].imshow(data[6], interpolation='none', cmap=colors)

axs[2, 0].set_ylabel("60 mm", fontsize='x-large')

axs[2, 1].imshow(data[7], interpolation='none', cmap=colors)

axs[2, 2].imshow(data[8], interpolation='none', cmap=colors)





fig.tight_layout()

plt.show()