xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update .pre-commit-config.yaml (#1026)

Browse Source
This commit is contained in:
Glenn Jocher
2023-02-17 22:26:40 +01:00
committed by GitHub
parent 9047d737f4
commit edd3ff1669
76 changed files with 928 additions and 935 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ultralytics/tracker/track.py
View File

@ -4,14 +4,14 @@ from ultralytics.tracker import BOTSORT, BYTETracker
from ultralytics.yolo.utils import IterableSimpleNamespace, yaml_load
from ultralytics.yolo.utils.checks import check_requirements, check_yaml
TRACKER_MAP = {"bytetrack": BYTETracker, "botsort": BOTSORT}
TRACKER_MAP = {'bytetrack': BYTETracker, 'botsort': BOTSORT}
check_requirements('lap') # for linear_assignment
def on_predict_start(predictor):
tracker = check_yaml(predictor.args.tracker)
cfg = IterableSimpleNamespace(**yaml_load(tracker))
assert cfg.tracker_type in ["bytetrack", "botsort"], \
assert cfg.tracker_type in ['bytetrack', 'botsort'], \
f"Only support 'bytetrack' and 'botsort' for now, but got '{cfg.tracker_type}'"
trackers = []
for _ in range(predictor.dataset.bs):
@ -38,5 +38,5 @@ def on_predict_postprocess_end(predictor):
def register_tracker(model):
model.add_callback("on_predict_start", on_predict_start)
model.add_callback("on_predict_postprocess_end", on_predict_postprocess_end)
model.add_callback('on_predict_start', on_predict_start)
model.add_callback('on_predict_postprocess_end', on_predict_postprocess_end)

4
ultralytics/tracker/trackers/byte_tracker.py
View File

@ -153,7 +153,7 @@ class STrack(BaseTrack):
return ret
def __repr__(self):
return f"OT_{self.track_id}_({self.start_frame}-{self.end_frame})"
return f'OT_{self.track_id}_({self.start_frame}-{self.end_frame})'
class BYTETracker:
@ -206,7 +206,7 @@ class BYTETracker:
strack_pool = self.joint_stracks(tracked_stracks, self.lost_stracks)
# Predict the current location with KF
self.multi_predict(strack_pool)
if hasattr(self, "gmc"):
if hasattr(self, 'gmc'):
warp = self.gmc.apply(img, dets)
STrack.multi_gmc(strack_pool, warp)
STrack.multi_gmc(unconfirmed, warp)

8
ultralytics/tracker/utils/gmc.py
View File

@ -50,14 +50,14 @@ class GMC:
seqName = seqName[:-6]
elif '-DPM' in seqName or '-SDP' in seqName:
seqName = seqName[:-4]
self.gmcFile = open(f"{filePath}/GMC-{seqName}.txt")
self.gmcFile = open(f'{filePath}/GMC-{seqName}.txt')
if self.gmcFile is None:
raise ValueError(f"Error: Unable to open GMC file in directory:{filePath}")
raise ValueError(f'Error: Unable to open GMC file in directory:{filePath}')
elif self.method in ['none', 'None']:
self.method = 'none'
else:
raise ValueError(f"Error: Unknown CMC method:{method}")
raise ValueError(f'Error: Unknown CMC method:{method}')
self.prevFrame = None
self.prevKeyPoints = None
@ -302,7 +302,7 @@ class GMC:
def applyFile(self, raw_frame, detections=None):
line = self.gmcFile.readline()
tokens = line.split("\t")
tokens = line.split('\t')
H = np.eye(2, 3, dtype=np.float_)
H[0, 0] = float(tokens[1])
H[0, 1] = float(tokens[2])