运行cfgs/lightstereo/lightstereo_s_sceneflow.yaml 报错

[songjiahao-wq]

(torch) E:\project2024\stereo matching\OpenStereo>python tools/infer.py --cfg_file cfgs/lightstereo/lightstereo_s_sceneflow.yaml --left_img_path rectified_left.png --right_img_path rectified_right.png
2025-02-26 14:40:22,688 INFO dist_mode False
2025-02-26 14:40:22,688 INFO cfg_file cfgs/lightstereo/lightstereo_s_sceneflow.yaml
2025-02-26 14:40:22,688 INFO left_img_path rectified_left.png
2025-02-26 14:40:22,688 INFO right_img_path rectified_right.png
2025-02-26 14:40:22,689 INFO pretrained_model None
2025-02-26 14:40:22,689 INFO savename None
2025-02-26 14:40:22,689 INFO run_mode infer
2025-02-26 14:40:22,689 INFO ----------- DATA_CONFIG -----------
2025-02-26 14:40:22,689 INFO cfgs.DATA_CONFIG.DATA_INFOS: [{'DATASET': 'SceneFlowDataset', 'DATA_SPLIT': {'TRAINING': './data/SceneFlow/sceneflow_finalpass_train.txt', 'EVALUATING': './data/SceneFlow/sceneflow_finalpass_test.txt', 'TESTING': './data/SceneFlow/sceneflow_finalpass_test.txt'}, 'DATA_PATH': '/yourpath/SceneFlow', 'RETURN_RIGHT_DISP': False}]
2025-02-26 14:40:22,689 INFO ----------- DATA_TRANSFORM -----------
2025-02-26 14:40:22,689 INFO cfgs.DATA_CONFIG.DATA_TRANSFORM.TRAINING: [{'NAME': 'RandomCrop', 'SIZE': [320, 736], 'Y_JITTER': False}, {'NAME': 'TransposeImage'}, {'NAME': 'ToTensor'}, {'NAME': 'NormalizeImage', 'MEAN': [0.485, 0.456, 0.406], 'STD': [0.229, 0.224, 0.225]}]
2025-02-26 14:40:22,689 INFO cfgs.DATA_CONFIG.DATA_TRANSFORM.EVALUATING: [{'NAME': 'RightTopPad', 'SIZE': [544, 960]}, {'NAME': 'TransposeImage'}, {'NAME': 'ToTensor'}, {'NAME': 'NormalizeImage', 'MEAN': [0.485, 0.456, 0.406], 'STD': [0.229, 0.224, 0.225]}]
2025-02-26 14:40:22,689 INFO ----------- MODEL -----------
2025-02-26 14:40:22,689 INFO cfgs.MODEL.NAME: LightStereo
2025-02-26 14:40:22,689 INFO cfgs.MODEL.MAX_DISP: 192
2025-02-26 14:40:22,689 INFO cfgs.MODEL.EXPANSE_RATIO: 4
2025-02-26 14:40:22,689 INFO cfgs.MODEL.AGGREGATION_BLOCKS: [1, 2, 4]
2025-02-26 14:40:22,689 INFO cfgs.MODEL.LEFT_ATT: True
2025-02-26 14:40:22,689 INFO cfgs.MODEL.FIND_UNUSED_PARAMETERS: False
2025-02-26 14:40:22,689 INFO cfgs.MODEL.CKPT: -1
2025-02-26 14:40:22,689 INFO cfgs.MODEL.PRETRAINED_MODEL:
2025-02-26 14:40:22,689 INFO ----------- OPTIMIZATION -----------
2025-02-26 14:40:22,689 INFO cfgs.OPTIMIZATION.FREEZE_BN: False
2025-02-26 14:40:22,689 INFO cfgs.OPTIMIZATION.SYNC_BN: True
2025-02-26 14:40:22,689 INFO cfgs.OPTIMIZATION.AMP: True
2025-02-26 14:40:22,690 INFO cfgs.OPTIMIZATION.BATCH_SIZE_PER_GPU: 24
2025-02-26 14:40:22,690 INFO cfgs.OPTIMIZATION.NUM_EPOCHS: 90
2025-02-26 14:40:22,690 INFO ----------- OPTIMIZER -----------
2025-02-26 14:40:22,690 INFO cfgs.OPTIMIZATION.OPTIMIZER.NAME: AdamW
2025-02-26 14:40:22,690 INFO cfgs.OPTIMIZATION.OPTIMIZER.LR: 0.0024
2025-02-26 14:40:22,690 INFO cfgs.OPTIMIZATION.OPTIMIZER.WEIGHT_DECAY: 1e-05
2025-02-26 14:40:22,690 INFO cfgs.OPTIMIZATION.OPTIMIZER.EPS: 1e-08
2025-02-26 14:40:22,690 INFO ----------- SCHEDULER -----------
2025-02-26 14:40:22,691 INFO cfgs.OPTIMIZATION.SCHEDULER.NAME: OneCycleLR
2025-02-26 14:40:22,691 INFO cfgs.OPTIMIZATION.SCHEDULER.MAX_LR: 0.0024
2025-02-26 14:40:22,691 INFO cfgs.OPTIMIZATION.SCHEDULER.PCT_START: 0.01
2025-02-26 14:40:22,692 INFO cfgs.OPTIMIZATION.SCHEDULER.ON_EPOCH: False
2025-02-26 14:40:22,692 INFO ----------- CLIP_GRAD -----------
2025-02-26 14:40:22,692 INFO cfgs.OPTIMIZATION.CLIP_GRAD.TYPE: value
2025-02-26 14:40:22,692 INFO cfgs.OPTIMIZATION.CLIP_GRAD.CLIP_VALUE: 0.1
2025-02-26 14:40:22,692 INFO ----------- EVALUATOR -----------
2025-02-26 14:40:22,692 INFO cfgs.EVALUATOR.BATCH_SIZE_PER_GPU: 12
2025-02-26 14:40:22,692 INFO cfgs.EVALUATOR.MAX_DISP: 192
2025-02-26 14:40:22,693 INFO cfgs.EVALUATOR.METRIC: ['d1_all', 'epe', 'thres_1', 'thres_2', 'thres_3']
2025-02-26 14:40:22,693 INFO ----------- TRAINER -----------
2025-02-26 14:40:22,693 INFO cfgs.TRAINER.EVAL_INTERVAL: 1
2025-02-26 14:40:22,693 INFO cfgs.TRAINER.CKPT_SAVE_INTERVAL: 1
2025-02-26 14:40:22,693 INFO cfgs.TRAINER.MAX_CKPT_SAVE_NUM: 30
2025-02-26 14:40:22,693 INFO cfgs.TRAINER.LOGGER_ITER_INTERVAL: 10
2025-02-26 14:40:22,693 INFO cfgs.TRAINER.TRAIN_VISUALIZATION: True
2025-02-26 14:40:22,694 INFO cfgs.TRAINER.EVAL_VISUALIZATION: True
Traceback (most recent call last):
File "E:\project2024\stereo matching\OpenStereo\tools\infer.py", line 92, in
main()
File "C:\ProgramData\anaconda3\envs\torch\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "E:\project2024\stereo matching\OpenStereo\tools\infer.py", line 82, in main
model_pred = model(sample)
File "C:\ProgramData\anaconda3\envs\torch\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "E:\project2024\stereo matching\OpenStereo.\stereo\modeling\models\lightstereo\lightstereo.py", line 48, in forward
features_left = self.backbone(image1)
File "C:\ProgramData\anaconda3\envs\torch\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "E:\project2024\stereo matching\OpenStereo.\stereo\modeling\models\lightstereo\backbone.py", line 68, in forward
p4 = self.fpn_layer4(c5, c4) # [bz, 96, H/16, W/16]
File "C:\ProgramData\anaconda3\envs\torch\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "E:\project2024\stereo matching\OpenStereo.\stereo\modeling\models\lightstereo\backbone.py", line 24, in forward
feat = torch.cat([high, low], 1)
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 49 but got size 50 for tensor number 1 in the list.