Pedro Henrique Conrado commited on
Commit
569b361
·
1 Parent(s): 5263ad0
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -21,7 +21,7 @@ import numpy as np
21
  import gradio as gr
22
  from tqdm import tqdm
23
  import yolox
24
- from yolox.tracker.byte_tracker import BYTETracker, STrack
25
 
26
 
27
  MODEL = "./best.pt"
 
21
  import gradio as gr
22
  from tqdm import tqdm
23
  import yolox
24
+ from bytetracker import BYTETracker, STrack
25
 
26
 
27
  MODEL = "./best.pt"
requirements.txt CHANGED
@@ -65,4 +65,5 @@ tensorboard
65
  pycocotools>=2.0.2
66
  onnx>=1.13.0
67
  onnxruntime>=1.12.0
68
- onnx-simplifier==0.4.10
 
 
65
  pycocotools>=2.0.2
66
  onnx>=1.13.0
67
  onnxruntime>=1.12.0
68
+ onnx-simplifier==0.4.10
69
+ bytetracker