diff --git a/scene/dataset_readers.py b/scene/dataset_readers.py index 189d709..789e4bf 100644 --- a/scene/dataset_readers.py +++ b/scene/dataset_readers.py @@ -173,7 +173,7 @@ def readColmapSceneInfo(path, images, eval, llffhold=8): # 读取并处理相机参数,转换为内部使用的格式 cam_infos_unsorted = readColmapCameras(cam_extrinsics=cam_extrinsics, cam_intrinsics=cam_intrinsics, images_folder=os.path.join(path, reading_dir)) # 根据图片名称对相机信息进行排序,以保证顺序一致性 - cam_infos = sorted(cam_infos_unsorted.copy(), key = lambda x : (x.image_path.split('/')[-2], x.image_name)) + cam_infos = sorted(cam_infos_unsorted.copy(), key = lambda x : (x.image_path.split('/')[-2], int(x.image_name))) # 根据是否为评估模式(eval),将相机分为训练集和测试集 # 如果为评估模式,根据llffhold参数(通常用于LLFF数据集)间隔选择测试相机 diff --git a/train.py b/train.py index f619188..901ca2b 100644 --- a/train.py +++ b/train.py @@ -230,10 +230,10 @@ if __name__ == "__main__": parser.add_argument("--test_iterations", nargs="+", type=int, default=[7_000, 30_000]) parser.add_argument("--save_iterations", nargs="+", type=int, default=[7_000, 30_000]) parser.add_argument("--quiet", action="store_true") - parser.add_argument("--checkpoint_iterations", nargs="+", type=int, default=[]) + parser.add_argument("--checkpoint_iterations", nargs="+", type=int, default=[30_000]) parser.add_argument("--start_checkpoint", type=str, default = None) - # 从命令行参数中解析出所有的参数值,并与上面的设置的参数一起存储到 Namespace 对象中,即args + # 从命令行参数中解析出所有的参数值,并与上面设置的参数一起存储到 Namespace 对象中,即args args = parser.parse_args(sys.argv[1:]) args.save_iterations.append(args.iterations)