switch model to tai's and add issues
This commit is contained in:
parent
2b0cfc4b72
commit
7044b1e588
2 changed files with 6 additions and 1 deletions
5
ISSUES.md
Normal file
5
ISSUES.md
Normal file
|
|
@ -0,0 +1,5 @@
|
||||||
|
# Issues
|
||||||
|
|
||||||
|
- It doesn't really care what pt file is included and it always use YOLO's model id, for example if id 1 is apple, it still say person. maybe extract class list from yolo's .pt somehow?
|
||||||
|
|
||||||
|
- Potential race condition issue when multiple camera try to init with the same unconverted model.
|
||||||
|
|
@ -380,7 +380,7 @@ def main_multi_window():
|
||||||
with separate OpenCV windows for each stream.
|
with separate OpenCV windows for each stream.
|
||||||
"""
|
"""
|
||||||
GPU_ID = 0
|
GPU_ID = 0
|
||||||
MODEL_PATH = "models/yolov8n.pt"
|
MODEL_PATH = "bangchak/models/frontal_detection_v5.pt"
|
||||||
|
|
||||||
# Load camera URLs from environment
|
# Load camera URLs from environment
|
||||||
camera_urls = []
|
camera_urls = []
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue