mirrors/yolov7
1
0
Fork 0
mirror of https://github.com/WongKinYiu/yolov7.git synced 2025-06-03 21:54:57 +08:00
Code Issues Projects Releases Wiki Activity

Update add_nms.py

Browse Source
This commit is contained in:
Alexey 2022-07-23 04:54:18 +03:00 committed by GitHub
parent a5b853c3fa
commit aae70703f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/add_nms.py
View File

@ -1,7 +1,11 @@
import numpy as np import numpy as np
import onnx import onnx
from onnx import shape_inference from onnx import shape_inference
import onnx_graphsurgeon as gs try:
import onnx_graphsurgeon as gs
except Exception as e:
print('Import onnx_graphsurgeon failure: %s' % e)
import logging import logging
LOGGER = logging.getLogger(__name__) LOGGER = logging.getLogger(__name__)