hofee/starvla_benchmark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update the image size for inference

Browse Source
This commit is contained in:
Zhu Juan
2026-06-12 14:36:13 +08:00
parent 87f2a2abfc
commit cbf0edfcae
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
starvla_inference_server.py
View File

@@ -129,7 +129,7 @@ class StarvlaInferenceServer:
def inference(self, observation: dict) -> dict: def inference(self, observation: dict) -> dict:
img_head, state_vec, prompt = \ img_head, state_vec, prompt = \
self.parse_observation(observation) self.parse_observation(observation, target_size=(410, 224))
vla_input = { vla_input = {
# "batch_images": [[img_left, img_right, img_wrist]], # "batch_images": [[img_left, img_right, img_wrist]],
"image": [img_head], "image": [img_head],