Update h5_to_db.py
Browse files- 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 |
-
|
|
|
|
| 94 |
from PIL import Image
|
| 95 |
img = Image.open(path)
|
| 96 |
width, height = img.size
|
| 97 |
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
cy = float(height / 2.0) # 960.0
|
| 103 |
-
k = 0.0 # 歪みなし
|
| 104 |
|
| 105 |
-
print(f"
|
| 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
|