ENSEEIHT/projet-probleme-inverse-3D
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

aaaa

Browse source
This commit is contained in:
gdamms 2023-01-30 10:26:06 +01:00
parent 705b610d68
commit 6e9d74759a
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/main.py
View file

@ -10,12 +10,12 @@ from rich.progress import track
from borders import update_border from borders import update_border
from fvi import fast_voxel_intersect from fvi import fast_voxel_intersect
VOXEL_SIZE = 5e-3 VOXEL_SIZE = 5e-2
X_MIN, X_MAX = 0.7, 1.3 X_MIN, X_MAX = -1.2, 1.2
Y_MIN, Y_MAX = -0.05, 0.05 Y_MIN, Y_MAX = -1.2, 1.2
Z_MIN, Z_MAX = -0.05, 0.05 Z_MIN, Z_MAX = -1.2, 1.2
nb_frame = 24 nb_frame = 25
points = np.array( points = np.array(
[ [
@ -35,10 +35,10 @@ proj_mats = []
frames = [] frames = []
for k in range(nb_frame): for k in range(nb_frame):
frame = cv2.imread(f"data/torus/masks/Image{k:04}.png", cv2.IMREAD_GRAYSCALE) frame = cv2.imread(f"data/cuby/masks/Image{k:04}.png", cv2.IMREAD_GRAYSCALE)
frames.append(cv2.imread(f"data/torus/images/Image{k:04}.png", cv2.IMREAD_GRAYSCALE)) frames.append(cv2.imread(f"data/cuby/images/Image{k:04}.png", cv2.IMREAD_GRAYSCALE))
with open(f"data/torus/cameras/{k:04d}.pickle", "rb") as file: with open(f"data/cuby/cameras/{k:04d}.pickle", "rb") as file:
matrices = pickle.load(file) matrices = pickle.load(file)
proj_mat = matrices["P"] proj_mat = matrices["P"]
proj_mats.append(proj_mat) proj_mats.append(proj_mat)
@ -69,9 +69,9 @@ for k in range(nb_frame):
# cv2.circle(frame, (cam_point[0], cam_point[1]), 2, (255, 0, 255)) # cv2.circle(frame, (cam_point[0], cam_point[1]), 2, (255, 0, 255))
# for k in range(nb_frame): # for k in range(nb_frame):
# frame = cv2.imread(f'/tmp/masks/Image{k:04}.png') # frame = cv2.imread(f'data/cuby/masks/Image{k:04}.png')
# with open(f"/tmp/cameras/{k:04d}.pickle", 'rb') as file: # with open(f"data/cuby/cameras/{k:04d}.pickle", 'rb') as file:
# proj_mat = pickle.load(file)["P"] # proj_mat = pickle.load(file)["P"]
# cam_points = proj_mat @ points.T # cam_points = proj_mat @ points.T