2024-11-21 22:39:22 +08:00
|
|
|
|
import glob
|
|
|
|
|
|
import os.path as osp
|
|
|
|
|
|
import json
|
2024-12-05 21:27:45 +08:00
|
|
|
|
import os
|
2024-11-21 22:39:22 +08:00
|
|
|
|
|
|
|
|
|
|
def write_json(data, output_path):
|
|
|
|
|
|
with open(output_path, "w") as f:
|
|
|
|
|
|
json.dump(data, f, indent=None, separators=(',', ':'))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def read_json(input):
|
|
|
|
|
|
with open(input, "r") as f:
|
|
|
|
|
|
data = json.load(f)
|
|
|
|
|
|
return data
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def read_img_paths(imgFolder):
|
|
|
|
|
|
imgPaths = []
|
|
|
|
|
|
for extension in ["jpg", "png", "jpeg", "bmp"]:
|
|
|
|
|
|
imgPaths += glob.glob(osp.join(imgFolder, "*.{}".format(extension)))
|
|
|
|
|
|
return imgPaths
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_output_folder(baseFolder, outputFolder):
|
|
|
|
|
|
folder = osp.join(baseFolder, outputFolder)
|
|
|
|
|
|
if not osp.exists(folder):
|
|
|
|
|
|
os.makedirs(folder)
|
2024-12-05 21:27:45 +08:00
|
|
|
|
return folder
|
|
|
|
|
|
|
|
|
|
|
|
class DataPaths:
|
|
|
|
|
|
def __init__(self, base_dir):
|
|
|
|
|
|
self.base_path = base_dir
|
|
|
|
|
|
# intri路径
|
|
|
|
|
|
self.intri_base = osp.join(self.base_path, "intri")
|
|
|
|
|
|
# 内参图片路径，按照cam0, cam1, cam2, ...组织子文件夹
|
|
|
|
|
|
self.intri_chessboard_data = osp.join(self.intri_base, "chessboard_data")
|
|
|
|
|
|
# 内参标定中的棋盘格角点检测可视化， 按照cam0, cam1, cam2, ...组织子文件夹
|
|
|
|
|
|
self.intri_chessboard_vis = osp.join(self.intri_base, "chessboard_vis")
|
|
|
|
|
|
# 需要畸变校正的图片路径，文件夹下存放cam0.jpg, cam1.jpg, cam2.jpg, ...只有图片
|
|
|
|
|
|
self.intri_undistort_data = osp.join(self.intri_base, "undistort_data")
|
|
|
|
|
|
self.intri_undistort_result = osp.join(self.intri_base, "undistort_result")
|
|
|
|
|
|
|
|
|
|
|
|
# extri路径
|
|
|
|
|
|
self.extri_base = osp.join(self.base_path, "extri")
|
|
|
|
|
|
# 外参标定使用的棋盘格图片路径，文件夹下存放cam0.jpg, cam1.jpg, cam2.jpg, ...一个相机一个图片
|
|
|
|
|
|
self.extri_chessboard_data = osp.join(self.extri_base,"extri_data")
|
|
|
|
|
|
# 绘制外参标定中的棋盘格角点检测可视化（原点和终点）， 按照cam0, cam1, cam2, ...组织子文件夹
|
|
|
|
|
|
self.extri_chessboard_vis = osp.join(self.extri_base, "extri_vis")
|
|
|
|
|
|
|
|
|
|
|
|
# check需要的路径
|
|
|
|
|
|
self.check_base = osp.join(self.base_path, "check")
|
|
|
|
|
|
# check使用的棋盘格图片路径，文件夹下存放cam0.jpg, cam1.jpg, cam2.jpg, ...
|
|
|
|
|
|
# 一个相机一张图片，主要是用于角点提取函数
|
|
|
|
|
|
self.check_data = osp.join(self.check_base, "check_data")
|
|
|
|
|
|
# 绘制外参标定中的棋盘格角点检测可视化（原点和终点）， 按照cam0, cam1, cam2, ...组织子文件夹
|
|
|
|
|
|
self.check_vis = osp.join(self.check_base, "extri_vis")
|
|
|
|
|
|
|
|
|
|
|
|
# json输出路径
|
|
|
|
|
|
self.json_out_base = osp.join(self.base_path, 'json_out')
|
|
|
|
|
|
self.intri_json_path = osp.join(self.json_out_base, 'intri.json')
|
|
|
|
|
|
self.extri_json_path = osp.join(self.json_out_base, 'extri.json')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def print_paths(self):
|
|
|
|
|
|
print("to be implemented")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
base_dir = "data"
|
|
|
|
|
|
DataPath = DataPaths(base_dir)
|
