Python cv2.TrackerCSRT_create() Examples
The following are 7
code examples of cv2.TrackerCSRT_create().
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example.
You may also want to check out all available functions/classes of the module
cv2
, or try the search function
.
Example #1
Source File: test_blob.py From ivy with MIT License | 7 votes |
def test_blob_update(): _bounding_box = [1, 1, 4, 4] _type = 'car' _confidence = 0.99 _tracker = cv2.TrackerKCF_create() blob = Blob(_bounding_box, _type, _confidence, _tracker) _new_bounding_box = [2, 2, 5, 5] _new_type = 'bus' _new_confidence = 0.35 _new_tracker = cv2.TrackerCSRT_create() blob.update(_new_bounding_box, _new_type, _new_confidence, _new_tracker) assert blob.bounding_box == _new_bounding_box assert blob.type == _new_type assert blob.type_confidence == _new_confidence assert blob.tracker == _new_tracker
Example #2
Source File: tracker.py From ivy with MIT License | 6 votes |
def _csrt_create(bounding_box, frame): ''' Create an OpenCV CSRT Tracker object. ''' tracker = cv2.TrackerCSRT_create() tracker.init(frame, tuple(bounding_box)) return tracker
Example #3
Source File: SpecificFaceDetector.py From robot-camera-platform with GNU General Public License v3.0 | 6 votes |
def process(self, image): self.detected = False self.__face_recognition.put_image(image) original_image, coordonates = self.__face_recognition.get_result() if original_image is not None: x, y, x1, y1 = coordonates if not hasattr(cv2, 'TrackerCSRT_create') or not callable(getattr(cv2, 'TrackerCSRT_create')): self.circle_coordonates = self.__get_center_radius((x, y, x1, y1)) self.detected = True return self.__tracker = cv2.TrackerCSRT_create() self.__tracker.init(original_image, (x, y, (x1 - x), abs(y1 - y))) return if self.__tracker is None: return (success, box) = self.__tracker.update(image) if success: (x, y, w, h) = [int(v) for v in box] self.circle_coordonates = self.__get_center_radius((x, y, x + w, y + h)) self.detected = True
Example #4
Source File: ground-control.py From aws-builders-fair-projects with Apache License 2.0 | 6 votes |
def predictionCallback(self, client, userdata, message): print('<< predictionCallback() >>') data = json.loads(message.payload.decode()) if len(data['prediction']) > 0 and data['prediction'][0][0] > -1: self.last_confidence = round(data['prediction'][0][2]*100,2) self.last_detected_class = int(data['prediction'][0][0]) x1 = data['prediction'][0][2]*self.IMAGE_WIDTH y1 = data['prediction'][0][3]*self.IMAGE_HEIGHT x2 = data['prediction'][0][4]*self.IMAGE_WIDTH y2 = data['prediction'][0][5]*self.IMAGE_HEIGHT w = x2-x1 h = y2-y1 print("({},{},{},{})".format(x1,x2,y1,y2)) print("({},{})".format(w,h)) self.initBB = (int(x1), int(y1), int(w), int(h)) # self.tracker = cv2.TrackerMOSSE_create() # self.tracker = cv2.TrackerKCF_create() self.tracker = cv2.TrackerCSRT_create() self.tracker.init(self.inferenceFrame, self.initBB) self.trackerInitialized = True
Example #5
Source File: main_auto.py From OpenLabeling with Apache License 2.0 | 5 votes |
def call_tracker_constructor(self, tracker_type): # -- TODO: remove this if I assume OpenCV version > 3.4.0 if int(self.major_ver == 3) and int(self.minor_ver) < 3: tracker = cv2.Tracker_create(tracker_type) # -- else: if tracker_type == 'CSRT': tracker = cv2.TrackerCSRT_create() elif tracker_type == 'KCF': tracker = cv2.TrackerKCF_create() elif tracker_type == 'MOSSE': tracker = cv2.TrackerMOSSE_create() elif tracker_type == 'MIL': tracker = cv2.TrackerMIL_create() elif tracker_type == 'BOOSTING': tracker = cv2.TrackerBoosting_create() elif tracker_type == 'MEDIANFLOW': tracker = cv2.TrackerMedianFlow_create() elif tracker_type == 'TLD': tracker = cv2.TrackerTLD_create() elif tracker_type == 'GOTURN': tracker = cv2.TrackerGOTURN_create() return tracker
Example #6
Source File: opencv_cftracker.py From pyCFTrackers with MIT License | 5 votes |
def init(self, first_frame, bbox): if self.name == 'KCF': self.tracker = cv2.TrackerKCF_create() elif self.name == 'MOSSE': self.tracker = cv2.TrackerMOSSE_create() elif self.name == 'CSRDCF': self.tracker = cv2.TrackerCSRT_create() else: raise NotImplementedError self.tracker.init(first_frame, bbox)
Example #7
Source File: main.py From OpenLabeling with Apache License 2.0 | 4 votes |
def call_tracker_constructor(self, tracker_type): if tracker_type == 'DASIAMRPN': tracker = dasiamrpn() else: # -- TODO: remove this if I assume OpenCV version > 3.4.0 if int(self.major_ver == 3) and int(self.minor_ver) < 3: #tracker = cv2.Tracker_create(tracker_type) pass # -- else: try: tracker = cv2.TrackerKCF_create() except AttributeError as error: print(error) print('\nMake sure that OpenCV contribute is installed: opencv-contrib-python\n') if tracker_type == 'CSRT': tracker = cv2.TrackerCSRT_create() elif tracker_type == 'KCF': tracker = cv2.TrackerKCF_create() elif tracker_type == 'MOSSE': tracker = cv2.TrackerMOSSE_create() elif tracker_type == 'MIL': tracker = cv2.TrackerMIL_create() elif tracker_type == 'BOOSTING': tracker = cv2.TrackerBoosting_create() elif tracker_type == 'MEDIANFLOW': tracker = cv2.TrackerMedianFlow_create() elif tracker_type == 'TLD': tracker = cv2.TrackerTLD_create() elif tracker_type == 'GOTURN': tracker = cv2.TrackerGOTURN_create() return tracker