import datasets import random import numpy import json import gradio import matplotlib.pyplot # for colormap import matplotlib.colors # for color conversion vr = datasets.load_dataset("cmudrc/porous-microstructure-strain-fields", data_files=['variable_radius.zip'], split='train') vn = datasets.load_dataset("cmudrc/porous-microstructure-strain-fields", data_files=['variable_number.zip'], split='train') circle = datasets.load_dataset("cmudrc/porous-microstructure-strain-fields", data_files=['circle/circle_test.zip'], split="train[:10]") crescent = datasets.load_dataset("cmudrc/porous-microstructure-strain-fields", data_files=['crescent/crescent_test.zip'], split="train[:10]") peanut = datasets.load_dataset("cmudrc/porous-microstructure-strain-fields", data_files=['peanut/peanut_test.zip'], split="train[:10]") ellipse = datasets.load_dataset("cmudrc/porous-microstructure-strain-fields", data_files=['ellipse/ellipse_test.zip'], split="train[:10]") triangle = datasets.load_dataset("cmudrc/porous-microstructure-strain-fields", data_files=['triangle/triangle_test.zip'], split="train[:10]") rectangle = datasets.load_dataset("cmudrc/porous-microstructure-strain-fields", data_files=['rectangle/rectangle_test.zip'], split="train[:10]") shapes = { "circle": circle, "crescent": crescent, "ellipse": ellipse, "peanut": peanut, "triangle": triangle, "rectangle": rectangle, } def randomize(selection): index = random.randint(0, 9) mask = 255*numpy.array(json.loads(shapes[selection]['Defects'][index])) v = numpy.array(json.loads(shapes[selection]['Strain'][index])) # Get the color map by name: cm = matplotlib.pyplot.get_cmap('RdBu') measure = max(v.max(), -v.min()) output = (v / measure) legend = "
{value:+.2e} | " legend = legend + "