mirror of https://github.com/WongKinYiu/yolov7.git
update readme (#396)
parent
c51c13a23f
commit
1e51f564e0
|
@ -169,8 +169,6 @@ wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-tiny.pt
|
|||
python export.py --weights ./yolov7-tiny.pt --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640
|
||||
git clone https://github.com/Linaom1214/tensorrt-python.git
|
||||
python ./tensorrt-python/export.py -o yolov7-tiny.onnx -e yolov7-tiny-nms.trt -p fp16
|
||||
|
||||
# example of inference in C++ https://github.com/Linaom1214/tensorrt-python/tree/main/yolov7/cpp
|
||||
```
|
||||
|
||||
**Pytorch to TensorRT another way** <a href="https://colab.research.google.com/gist/AlexeyAB/fcb47ae544cf284eb24d8ad8e880d45c/yolov7trtlinaom.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a> <details><summary> <b>Expand</b> </summary>
|
||||
|
|
Loading…
Reference in New Issue