When run the test_model.py, the checkpoint loader seem to be wrong, the error imformation is below.

[RongkunYang]

File "/data1/program/digital-cousins/tests/test_models.py", line 228, in main(args) File "/data1/program/digital-cousins/tests/test_models.py", line 201, in main test_perspective_fields(args) File "/data1/program/digital-cousins/tests/test_models.py", line 36, in test_perspective_fields PerspectiveFields().estimate_camera_intrinsics(input_path=TEST_IMG_PATH) File "/data1/program/digital-cousins/digital_cousins/models/perspective_fields.py", line 36, in init self.model = _PerspectiveFields(version) File "/data1/program/digital-cousins/deps/PerspectiveFields/perspective2d/perspectivefields.py", line 163, in init self._init_weights() File "/data1/program/digital-cousins/deps/PerspectiveFields/perspective2d/perspectivefields.py", line 181, in _init_weights state_dict = torch.hub.load_state_dict_from_url( File "/home/yrk/anaconda3/envs/acdc/lib/python3.10/site-packages/torch/hub.py", line 871, in load_state_dict_from_url return torch.load(cached_file, map_location=map_location, weights_only=weights_only) File "/home/yrk/anaconda3/envs/acdc/lib/python3.10/site-packages/torch/serialization.py", line 1326, in load with _open_zipfile_reader(opened_file) as opened_zipfile: File "/home/yrk/anaconda3/envs/acdc/lib/python3.10/site-packages/torch/serialization.py", line 671, in init super().init(torch._C.PyTorchFileReader(name_or_buffer)) RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

[RongkunYang]

My Pytorch version is 2.5, and the checkpoint seem to be download completely, so may I ask how can the error can be fixed?

[RongkunYang]

OK, I download the perspective checkpoint manually, the problem seem to be resolved.

[cremebrule]

Great!