Skip to content

Commit

Permalink
not important changes
Browse files Browse the repository at this point in the history
Juanma committed Dec 9, 2022

Verified

This commit was signed with the committer’s verified signature.
redeboer Remco de Boer
1 parent 6e45ea5 commit 81f507b
Showing 2 changed files with 23 additions and 16 deletions.
2 changes: 1 addition & 1 deletion environment.yml
Original file line number Diff line number Diff line change
@@ -19,5 +19,5 @@ dependencies:
- mayavi
- pip
- pip:
- tensorflow==2.7.0
- tensorflow==2.9.0

37 changes: 22 additions & 15 deletions src/utils/view_edm.py
Original file line number Diff line number Diff line change
@@ -35,20 +35,26 @@
# # host = pickle.load(pfile)[8:-8, 8:-8, 8:-8]
# # output.view_with_mayavi(grid.x, grid.y, grid.z, host)

# host = np.float32(host)
# host = np.expand_dims(host, axis=[0, -1])
# # output.view_with_mayavi(grid.x, grid.y, grid.z, host[0,:,:,:,0])
# datap = max_pool3d(host, 2, 1, 'SAME')
# datan = max_pool3d(host*-1, 2, 1, 'SAME')
# host = datap + datan*-1
# # output.view_with_mayavi(grid.x, grid.y, grid.z, host[0, :, :, :, 0])

with open("cage_esp_optimized_final_ESP.p", 'rb') as pfile:
esp = pickle.load(pfile)
datap = max_pool3d(esp, 3, 1, 'SAME')
datan = max_pool3d(esp*-1, 3, 1, 'SAME')
esp = datap + datan*-1
esp = esp / 0.33
# with open("cc6_esp.pkl", 'rb') as pfile:
# host = pickle.load(pfile) #* -1

with open("pdcage_esp.pkl", 'rb') as pfile:
host = pickle.load(pfile)[8:-8, 8:-8, 8:-8] #* -1

host = np.float32(host)
host = np.expand_dims(host, axis=[0, -1])
# output.view_with_mayavi(grid.x, grid.y, grid.z, host[0,:,:,:,0])
datap = max_pool3d(host, 5, 1, 'SAME')
datan = max_pool3d(host*-1, 5, 1, 'SAME')
host = datap*1 + datan*-1
output.view_with_mayavi(grid.x, grid.y, grid.z, host[0, :, :, :, 0])

# with open("cage_esp_optimized_final_ESP.p", 'rb') as pfile:
# esp = pickle.load(pfile)
# datap = max_pool3d(esp, 3, 1, 'SAME')
# datan = max_pool3d(esp*-1, 3, 1, 'SAME')
# esp = datap + datan*-1
# esp = esp / 0.33


if len(cubes) == 2:
@@ -77,4 +83,5 @@
else:
for i in range(args.render):
print(i+1)
output.view_with_mayavi(grid.x, grid.y, grid.z, esp[1]*cubes[1][:, :, :, 0])
# output.view_with_mayavi(grid.x, grid.y, grid.z, esp[1]*cubes[1][:, :, :, 0])
output.view_with_mayavi(grid.x, grid.y, grid.z, host+cubes[i][:, :, :, 0])

0 comments on commit 81f507b

Please sign in to comment.