Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
ojh6404 committed Jan 13, 2024
1 parent a49e00c commit 46a90ab
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion node_scripts/deva_node.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ def callback(self, img_msg):
self.image = self.bridge.imgmsg_to_cv2(img_msg, desired_encoding="rgb8")
with torch.cuda.amp.autocast(enabled=self.cfg["amp"]):
min_size = min(self.image.shape[:2])
deva_input = get_input_frame_for_deva(self.image, min_size)
deva_input = get_input_frame_for_deva(self.image, min_size).to(self.deva_config.device)
if self.cnt % self.cfg["detection_every"] == 0:
incorporate_mask, segments_info = segment_with_text(
self.cfg,
Expand Down

0 comments on commit 46a90ab

Please sign in to comment.