Skip to content

Commit

Permalink
[mir_perception][mir_object_recognition] remove YOLOv7 imports
Browse files Browse the repository at this point in the history
  • Loading branch information
kvnptl authored Jul 24, 2023
1 parent d9595ec commit 55d0f91
Showing 1 changed file with 0 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -9,16 +9,8 @@ import torch
from cv_bridge import CvBridge, CvBridgeError
from mas_perception_msgs.msg import ImageList, Object, ObjectList
from sensor_msgs.msg import Image, RegionOfInterest
from rgb_object_recognition.yolov7.models.experimental import attempt_load
from rgb_object_recognition.yolov7.utils.datasets import letterbox
from rgb_object_recognition.yolov7.utils.general import check_img_size, non_max_suppression, apply_classifier, \
scale_coords, xyxy2xywh
from rgb_object_recognition.yolov7.utils.plots import plot_one_box
from rgb_object_recognition.yolov7.utils.torch_utils import select_device, TracedModel, load_classifier, \
time_synchronized
from ultralytics import YOLO


class RGBObjectRecognizer():
def __init__(self, weights,
net='detection',
Expand Down

0 comments on commit 55d0f91

Please sign in to comment.