Skip to content

Commit

Permalink
[format] isort
Browse files Browse the repository at this point in the history
  • Loading branch information
mqcmd196 committed Feb 5, 2025
1 parent 04a5449 commit 65fbf26
Show file tree
Hide file tree
Showing 5 changed files with 13 additions and 19 deletions.
4 changes: 2 additions & 2 deletions example/client.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
#!/usr/bin/env python3
import argparse

import cv2
import matplotlib.pyplot as plt
import rospy
from cv_bridge import CvBridge

from detic_ros.srv import DeticSeg

import cv2

if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('-image', type=str, help='input image path')
Expand Down
3 changes: 1 addition & 2 deletions example/masked_image_publisher.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,8 @@
import numpy as np
import rospy
from cv_bridge import CvBridge
from sensor_msgs.msg import Image

from detic_ros.msg import SegmentationInfo
from sensor_msgs.msg import Image


class SampleNode:
Expand Down
3 changes: 2 additions & 1 deletion node_script/batch_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,10 @@
import rosbag
import tqdm
from cv_bridge import CvBridge
from sensor_msgs.msg import Image

from moviepy.editor import ImageSequenceClip
from node_config import NodeConfig
from sensor_msgs.msg import Image
from wrapper import DeticWrapper, InferenceRawResult


Expand Down
19 changes: 7 additions & 12 deletions node_script/node.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,23 +2,18 @@
from typing import Optional

import rospy
import torch
from jsk_recognition_msgs.msg import LabelArray, VectorArray
from detic_ros.msg import SegmentationInfo
from detic_ros.node_config import NodeConfig
from detic_ros.srv import (CustomVocabulary, CustomVocabularyRequest,
CustomVocabularyResponse, DeticSeg, DeticSegRequest,
DeticSegResponse)
from detic_ros.wrapper import DeticWrapper
from jsk_recognition_msgs.msg import LabelArray, VectorArray
from rospy import Publisher, Subscriber
from sensor_msgs.msg import Image
from std_srvs.srv import Empty, EmptyRequest, EmptyResponse
from detic_ros.wrapper import DeticWrapper

from detic_ros.msg import SegmentationInfo
from detic_ros.srv import (
CustomVocabulary,
CustomVocabularyRequest,
CustomVocabularyResponse,
DeticSeg,
DeticSegRequest,
DeticSegResponse,
)
import torch


class DeticRosNode:
Expand Down
3 changes: 1 addition & 2 deletions test/test_detic_ros_node.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,8 @@

import rospy
import rostest
from sensor_msgs.msg import Image

from detic_ros.msg import SegmentationInfo
from sensor_msgs.msg import Image


class TestNode(unittest.TestCase):
Expand Down

0 comments on commit 65fbf26

Please sign in to comment.