thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #150 from JWLee89/jaylee

Browse Source 
Updated gen_wts_yoloV5.py - cfg file fails to write due to encoding issues
This commit is contained in:
Marcos Luciano
2022-03-09 08:57:18 -03:00
committed by GitHub
parent f8222c9665 bfc97db42c
commit adf8168fb5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/gen_wts_yoloV5.py
View File

@@ -124,7 +124,7 @@ with open(wts_file, "w") as f:
f.write(wts_write) f.write(wts_write)
with open(cfg_file, "w") as c: with open(cfg_file, "w") as c:
with open(yaml_file, "r") as f: with open(yaml_file, "r", encoding="utf-8") as f:
nc = 0 nc = 0
depth_multiple = 0 depth_multiple = 0
width_multiple = 0 width_multiple = 0