Skip to content

Commit

Permalink
export images working and tested
Browse files Browse the repository at this point in the history
  • Loading branch information
@mattclifford1
mattclifford1 committed Jul 18, 2024
1 parent 7b7860f commit 8725293
Show file tree
Hide file tree
Showing 3 changed files with 78 additions and 31 deletions.
57 changes: 28 additions & 29 deletions IQM_Vis/UI/experiment_mode.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@
import IQM_Vis
from IQM_Vis.UI.custom_widgets import ClickLabel
from IQM_Vis.UI import utils
from IQM_Vis.utils import gui_utils, plot_utils, image_utils
from IQM_Vis.utils import gui_utils, plot_utils, image_utils, save_utils


class make_experiment(QMainWindow):
Expand All @@ -40,7 +40,7 @@ def __init__(self,
image_display_size,
rgb_brightness,
display_brightness,
default_save_dir=IQM_Vis.utils.save_utils.DEFAULT_SAVE_DIR,
default_save_dir=save_utils.DEFAULT_SAVE_DIR,
image_preprocessing='None',
image_postprocessing='None',
checked_metrics={}):
Expand Down Expand Up @@ -122,7 +122,7 @@ def show_all_images(self, tab='setup'):
ax.imshow(image_utils.calibrate_brightness(
trans['image'], self.rgb_brightness, self.display_brightness, ubyte=False))
ax.axis('off')
ax.set_title(make_name_for_trans(trans), fontsize=6)
ax.set_title(save_utils.make_name_for_trans(trans), fontsize=6)
# self.widget_experiments[tab]['images'].figure.tight_layout()

# time.sleep(5)
Expand All @@ -147,7 +147,8 @@ def get_all_images(self):
param = single_trans[trans_name]
data = {'transform_name': trans_name,
'transform_value': param}
self.original_params_order.append(make_name_for_trans(data))
self.original_params_order.append(
save_utils.make_name_for_trans(data))

# REFERENCE image
self.ref_image = self.data_store.get_reference_image()
Expand Down Expand Up @@ -196,7 +197,7 @@ def get_metric_scores(self):
for name, score in score_dict.items():
metrics.append(name)
scores.append(float(score))
IQM_scores[make_name_for_trans(data)] = scores
IQM_scores[save_utils.make_name_for_trans(data)] = scores
IQM_scores['IQM'] = metrics
self.IQM_scores_df = pd.DataFrame.from_dict(IQM_scores)
self.IQM_scores_df.set_index('IQM', inplace=True)
Expand Down Expand Up @@ -425,16 +426,16 @@ def save_experiment(self):
exp_save_dir = f'{self.default_save_dir}-experiment-{i}'
if os.path.exists(exp_save_dir):
# get transform funcs and params
exp_trans_params = IQM_Vis.utils.save_utils.load_obj(
exp_trans_params = save_utils.load_obj(
os.path.join(exp_save_dir, 'transforms', 'transform_params.pkl'))
exp_trans_funcs = IQM_Vis.utils.save_utils.load_obj(
exp_trans_funcs = save_utils.load_obj(
os.path.join(exp_save_dir, 'transforms', 'transform_functions.pkl'))

# get image processing saved params
processing_file = IQM_Vis.utils.save_utils.get_image_processing_file(exp_save_dir)
processing_file = save_utils.get_image_processing_file(exp_save_dir)
procesing_same = False
if os.path.exists(processing_file):
processing = IQM_Vis.utils.save_utils.load_json_dict(processing_file)
processing = save_utils.load_json_dict(processing_file)
if processing == self.processing:
procesing_same = True

Expand All @@ -454,34 +455,34 @@ def save_experiment(self):
os.makedirs(os.path.join(self.default_save_dir, 'transforms'), exist_ok=True)

# save experiment images
if not os.path.exists(IQM_Vis.utils.save_utils.get_original_image_file(self.default_save_dir)):
if not os.path.exists(save_utils.get_original_image_file(self.default_save_dir)):
image_utils.save_image(self.ref_image,
IQM_Vis.utils.save_utils.get_original_image_file(self.default_save_dir))
if not os.path.exists(IQM_Vis.utils.save_utils.get_original_unprocessed_image_file(self.default_save_dir)):
save_utils.get_original_image_file(self.default_save_dir))
if not os.path.exists(save_utils.get_original_unprocessed_image_file(self.default_save_dir)):
if hasattr(self, 'ref_image_unprocessed'):
image_utils.save_image(self.ref_image_unprocessed,
IQM_Vis.utils.save_utils.get_original_unprocessed_image_file(self.default_save_dir))
save_utils.get_original_unprocessed_image_file(self.default_save_dir))
if new_dir == True:
for trans in self.experiment_transforms:
image_utils.save_image(
trans['image'], os.path.join(self.default_save_dir, 'images', f'{make_name_for_trans(trans)}.png'))
trans['image'], os.path.join(self.default_save_dir, 'images', f'{save_utils.make_name_for_trans(trans)}.png'))
# save the transformations
IQM_Vis.utils.save_utils.save_obj(
IQM_Vis.utils.save_utils.get_transform_params_file(self.default_save_dir),
save_utils.save_obj(
save_utils.get_transform_params_file(self.default_save_dir),
self.original_params_order)
IQM_Vis.utils.save_utils.save_obj(
IQM_Vis.utils.save_utils.get_transform_functions_file(self.default_save_dir),
save_utils.save_obj(
save_utils.get_transform_functions_file(self.default_save_dir),
dict(sorted(trans_funcs.items())))
# save the image pre/post processing options
IQM_Vis.utils.save_utils.save_json_dict(
IQM_Vis.utils.save_utils.get_image_processing_file(self.default_save_dir),
save_utils.save_json_dict(
save_utils.get_image_processing_file(self.default_save_dir),
self.processing)

# save the experiment results
exp_order = []
for trans in self.experiment_transforms:
exp_order.append(make_name_for_trans(trans))
csv_file = IQM_Vis.utils.save_utils.save_experiment_results(
exp_order.append(save_utils.make_name_for_trans(trans))
csv_file = save_utils.save_experiment_results(
self.original_params_order,
exp_order,
self.default_save_dir,
Expand Down Expand Up @@ -559,7 +560,7 @@ def clicked_image(self, image_name, widget_name):
self.able_to_click = False
# get comparison to pivot
trans_str = image_name[len(self.image_name)+1:]
if trans_str != make_name_for_trans(self.pivot): # lower value
if trans_str != save_utils.make_name_for_trans(self.pivot): # lower value
# If element smaller than pivot is found swap it with the greater element pointed by i
self.less_than_pivot = True
else:
Expand Down Expand Up @@ -588,10 +589,12 @@ def change_experiment_images(self, A_trans, B_trans):

gui_utils.change_im(self.widget_experiments['exp']['A']['data'], A, resize=self.image_display_size,
rgb_brightness=self.rgb_brightness, display_brightness=self.display_brightness)
self.widget_experiments['exp']['A']['data'].setObjectName(f'{self.image_name}-{make_name_for_trans(A_trans)}')
self.widget_experiments['exp']['A']['data'].setObjectName(
f'{self.image_name}-{save_utils.make_name_for_trans(A_trans)}')
gui_utils.change_im(self.widget_experiments['exp']['B']['data'], B, resize=self.image_display_size,
rgb_brightness=self.rgb_brightness, display_brightness=self.display_brightness)
self.widget_experiments['exp']['B']['data'].setObjectName(f'{self.image_name}-{make_name_for_trans(B_trans)}')
self.widget_experiments['exp']['B']['data'].setObjectName(
f'{self.image_name}-{save_utils.make_name_for_trans(B_trans)}')

''' UI '''
def init_style(self, style='light', css_file=None):
Expand Down Expand Up @@ -662,7 +665,3 @@ def sort_list(list1, list2):
for i in inds:
sorted_list1.append(list1[i])
return sorted_list1

def make_name_for_trans(trans):
splitter = '-----'
return f"{trans['transform_name']}{splitter}{trans['transform_value']}"
47 changes: 45 additions & 2 deletions IQM_Vis/UI/widgets.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,9 @@
# License: BSD 3-Clause License

import re
import os
from functools import partial
import datetime

import numpy as np
from PyQt6.QtWidgets import QPushButton, QLabel, QSlider, QCheckBox, QComboBox, QLineEdit
Expand All @@ -14,7 +16,7 @@

import IQM_Vis
from IQM_Vis.UI.custom_widgets import ClickLabel
from IQM_Vis.utils import gui_utils, plot_utils, image_utils
from IQM_Vis.utils import gui_utils, plot_utils, image_utils, save_utils

# sub class used by IQM_Vis.main.make_app to initialise widgets and general UI functions for widgets
class widgets():
Expand Down Expand Up @@ -627,7 +629,48 @@ def change_text_export_trans(self, trans):
self.widget_export[trans][widget].setStyleSheet(f"QLineEdit {{color: gray;}}\nQLabel {{color: gray;}}")

def export_trans_images(self):
pass
# make save folder
image_name = f"{self.data_stores[0].get_reference_image_name()}-export"

save_folder = os.path.join(
self.default_save_dir, image_name, str(datetime.datetime.now()).split('.')[0])
if not os.path.exists(save_folder):
os.makedirs(save_folder)

# save original image
ref_image = self.data_stores[0].get_reference_image()
image_utils.save_image(ref_image,
os.path.join(save_folder, f"reference.png"))

# make and save transforms
checked_transforms = {}
for trans in self.widget_export:
if self.widget_export[trans]['check_box'].isChecked():
data = {'min': make_float_from_text(self.widget_export[trans]['min_edit'].text()),
'max': make_float_from_text(self.widget_export[trans]['max_edit'].text()),
'num_steps': int(self.widget_export[trans]['steps_edit'].text()),
'function': self.checked_transformations[trans]['function']
}
name = self.widget_export[trans]['name'].text()
checked_transforms[name] = data

export_images = plot_utils.get_all_single_transform_params(
checked_transforms, num_steps='from_dict')

# remove any params with value 0
export_images = [x for x in export_images if not x[list(x.keys())[0]] == 0]

# make and save images
for single_trans in export_images:
trans_name = list(single_trans.keys())[0]
param = single_trans[trans_name]
img = image_utils.get_transform_image(self.data_stores[0],
transform_functions={trans_name: self.checked_transformations[trans_name]},
transform_params={trans_name: param})
trans_info = {'transform_name': trans_name, 'transform_value': param}
image_utils.save_image(img, os.path.join(
save_folder, f'{save_utils.make_name_for_trans(trans_info)}.png'))
self.status_bar.showMessage(f'Images saved to {save_folder}', 5000)

'''
experiments
Expand Down
5 changes: 5 additions & 0 deletions IQM_Vis/utils/save_utils.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -120,3 +120,8 @@ def save_json_dict(path, dict_):
def load_json_dict(path):
with open(path, 'r') as fp:
return json.load(fp)


def make_name_for_trans(trans):
splitter = '-----'
return f"{trans['transform_name']}{splitter}{trans['transform_value']}"

0 comments on commit 8725293

Please sign in to comment.