added slice

app.py

@@ -4,6 +4,8 @@ from dcgan import DCGAN3D_G
 import torch
 import requests
 import time
+import numpy as np
+import streamlit.components.v1 as components
 
 url = "https://github.com/LukasMosser/PorousMediaGan/blob/master/checkpoints/berea/berea_generator_epoch_24.pth?raw=true"
 
@@ -14,7 +16,6 @@ with open('berea_generator_epoch_24.pth', 'wb') as f:
     f.write(resp.content)
     time.sleep(5)
 
-print(resp.status_code)
 st.text(resp.status_code)
 
 pv.set_plot_theme("document")
@@ -23,30 +24,32 @@ pl = pv.Plotter(shape=(1, 1),
 
 netG = DCGAN3D_G(64, 512, 1, 32, 1)
 netG.load_state_dict(torch.load("berea_generator_epoch_24.pth", map_location=torch.device('cpu')))
-z = torch.randn(1, 512, 3, 3, 3)
+z = torch.randn(1, 512, 1, 1, 1)
 with torch.no_grad():
     X = netG(z)
 
 st.image((X[0, 0, 32].numpy()+1)/2, output_format="png")
-"""
-data = examples.load_channels()
-channels = data.threshold([0.9, 1.1])
-print(channels)
-bodies = channels.split_bodies()
-# Now remove all bodies with a small volume
-for key in bodies.keys():
-    b = bodies[key]
-    vol = b.volume
-    if vol < 1000.0:
-        del bodies[key]
-        continue
-    # Now lets add a volume array to all blocks
-    b.cell_data["TOTAL VOLUME"] = np.full(b.n_cells, vol)
-
-for i, body in enumerate(bodies):
-    print(f"Body {i:02d} volume: {body.volume:.3f}")
-
-pl.add_mesh(bodies)
+
+img = (X[0, 0].numpy()+1)/2
+
+a = 0.9
+
+# create a uniform grid to sample the function with
+x_min, y_min, z_min = 0, 0, 0
+grid = pv.UniformGrid(
+    dims=img.shape,
+    spacing=(1, 1, 1),
+    origin=(x_min, y_min, z_min),
+)
+x, y, z = grid.points.T
+
+# sample and plot
+values = img.flatten()
+grid.point_data['my_array'] = values
+slices = grid.slice_orthogonal()
+
+pl.add_mesh(slices, cmap="gray")
+
 pl.export_html('pyvista.html')
 
 st.header("test html import")
@@ -56,7 +59,3 @@ HtmlFile = open("pyvista.html", 'r', encoding='utf-8')
 source_code = HtmlFile.read()
 
 components.html(source_code, width=view_width, height=view_height)
-
-#snippet = embed.embed_snippet(views=view(reader.GetOutput()))
-#html = embed.html_template.format(title="", snippet=snippet)
-#components.html(html, width=view_width, height=view_height)"""