Skip to content

Commit

Permalink
not important changes
Browse files Browse the repository at this point in the history
  • Loading branch information
Juanma committed Dec 9, 2022
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.