1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
| model_inputs = { "road": { "input_names": ['input_imgs', 'big_input_imgs', 'desire', 'traffic_convention', 'nav_features', 'nav_instructions', 'features_buffer'], "input_shapes": { 'input_imgs': [1, 12, 128, 256], 'big_input_imgs': [1, 12, 128, 256], 'desire': [1, 100, 8], 'traffic_convention': [1, 2], 'nav_features': [1, 256], 'nav_instructions': [1,150], 'features_buffer': [1, 99, 128] }, "input_dtypes": { 'input_imgs': np.float32, 'big_input_imgs': np.float32, 'desire': np.float32, 'traffic_convention': np.float32, 'nav_features': np.float32, 'nav_instructions': np.float32, 'features_buffer': np.float32 } }, "nav":{ "input_names": ['input_img'], "input_shapes": { "input_img":[1,1,256,256] }, "input_dtypes": { "input_img":np.float32 } }, "dmonitor":{ "input_names": ['input_img', 'calib'], "input_shapes": { "input_img":[1,1382400], "calib": [1,3] }, "input_dtypes": { "input_img":np.float32, "calib": np.float32 } } }
|