stpete2 commited on
Commit
ba6e567
·
verified ·
1 Parent(s): 913cb45

Update h5_to_db.py

Browse files
Files changed (1) hide show
  1. h5_to_db.py +7 -8
h5_to_db.py CHANGED
@@ -90,19 +90,18 @@ def add_keypoints(db, h5_path, image_path, img_ext, camera_model, single_camera=
90
  path = all_images[filename]
91
  fname_with_ext = os.path.basename(path)
92
 
93
- if camera_id is None or not single_camera:
 
94
  from PIL import Image
95
  img = Image.open(path)
96
  width, height = img.size
97
 
98
- # 適切な初期値を設定
99
- # あなたの画像: Width=1440, Height=1920
100
- focal_length = float(max(width, height)) # 1920
101
- cx = float(width / 2.0) # 720.0
102
- cy = float(height / 2.0) # 960.0
103
- k = 0.0 # 歪みなし
104
 
105
- print(f"カメラパラメータ: f={focal_length}, cx={cx}, cy={cy}, k={k}")
106
 
107
  camera_id = db.add_camera(
108
  model=2, # SIMPLE_RADIAL
 
90
  path = all_images[filename]
91
  fname_with_ext = os.path.basename(path)
92
 
93
+ # カメラは1回だけ作成(single_camera=Trueの場合)
94
+ if camera_id is None:
95
  from PIL import Image
96
  img = Image.open(path)
97
  width, height = img.size
98
 
99
+ focal_length = float(max(width, height))
100
+ cx = float(width / 2.0)
101
+ cy = float(height / 2.0)
102
+ k = 0.0
 
 
103
 
104
+ print(f"カメラ作成: f={focal_length}, cx={cx}, cy={cy}, k={k}")
105
 
106
  camera_id = db.add_camera(
107
  model=2, # SIMPLE_RADIAL