Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
nmhaddad authored Feb 20, 2024
1 parent 3f6785f commit 0488b1e
Showing 1 changed file with 8 additions and 9 deletions.
17 changes: 8 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,17 +27,16 @@ Package is installable with Python 3.9, and 3.10
from fast_track.databases import SQLDatabase
if __name__ == '__main__':
with open('config/coco.yml', 'r') as f:
config = yaml.safe_load(f)
with open('config/coco.yml', 'r') as f:
config = yaml.safe_load(f)
camera = cv2.VideoCapture(config['data_path'])
detector = YOLONAS(**config['detector'], names=config['names'], image_shape=(camera.get(3), camera.get(4)))
tracker = BYTETracker(**config['tracker'], names=config['names'])
database = SQLDatabase(**config["db"], class_names=config['names'])
camera = cv2.VideoCapture(config['data_path'])
detector = YOLONAS(**config['detector'], names=config['names'], image_shape=(camera.get(3), camera.get(4)))
tracker = BYTETracker(**config['tracker'], names=config['names'])
database = SQLDatabase(**config["db"], class_names=config['names'])
with Pipeline(camera=camera, detector=detector, tracker=tracker, database=database, outfile=config['outfile']) as p:
p.run()
with Pipeline(camera=camera, detector=detector, tracker=tracker, database=database, outfile=config['outfile']) as p:
p.run()
```
## Run the Demo
Expand Down

0 comments on commit 0488b1e

Please sign in to comment.