Skip to content

Commit c9dfbf2

Browse files
committed
update readme
1 parent f088ec7 commit c9dfbf2

File tree

5 files changed

+5
-5
lines changed

5 files changed

+5
-5
lines changed

yolov3-spp/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
Currently this is supporting dynamic input shape, if you want to use non-dynamic version, please checkout commit [659fd2b](https://github.com/wang-xinyu/tensorrtx/commit/659fd2b23482197b19dccf746a5a3dbff1611381).
44

5-
The Pytorch implementation is [ultralytics/yolov3](https://github.com/ultralytics/yolov3). It provides two trained weights of yolov3-spp, `yolov3-spp.pt` and `yolov3-spp-ultralytics.pt`(originally named `ultralytics68.pt`).
5+
The Pytorch implementation is [ultralytics/yolov3 archive branch](https://github.com/ultralytics/yolov3/tree/archive). It provides two trained weights of yolov3-spp, `yolov3-spp.pt` and `yolov3-spp-ultralytics.pt`(originally named `ultralytics68.pt`).
66

77
## Config
88

yolov3-tiny/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# yolov3-tiny
22

3-
The Pytorch implementation is [ultralytics/yolov3](https://github.com/ultralytics/yolov3).
3+
The Pytorch implementation is [ultralytics/yolov3 archive branch](https://github.com/ultralytics/yolov3/tree/archive).
44

55
## Excute:
66

yolov3/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# yolov3
22

3-
The Pytorch implementation is [ultralytics/yolov3](https://github.com/ultralytics/yolov3). It provides two trained weights of yolov3, `yolov3.weights` and `yolov3.pt`
3+
The Pytorch implementation is [ultralytics/yolov3 archive branch](https://github.com/ultralytics/yolov3/tree/archive). It provides two trained weights of yolov3, `yolov3.weights` and `yolov3.pt`
44

55
This branch is using tensorrt7 API, there is also a yolov3 implementation using tensorrt4 API, go to [branch trt4/yolov3](https://github.com/wang-xinyu/tensorrtx/tree/trt4/yolov3), which is using [ayooshkathuria/pytorch-yolo-v3](https://github.com/ayooshkathuria/pytorch-yolo-v3).
66

yolov4/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# yolov4
22

3-
The Pytorch implementation is from [ultralytics/yolov3](https://github.com/ultralytics/yolov3). It can load yolov4.cfg and yolov4.weights(from AlexeyAB/darknet).
3+
The Pytorch implementation is from [ultralytics/yolov3 archive branch](https://github.com/ultralytics/yolov3/tree/archive). It can load yolov4.cfg and yolov4.weights(from AlexeyAB/darknet).
44

55
## Config
66

yolov5/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -41,11 +41,11 @@ python gen_wts.py yolov5s.pt
4141
2. build tensorrtx/yolov5 and run
4242

4343
```
44-
cp {ultralytics}/yolov5/yolov5s.wts {tensorrtx}/yolov5/
4544
cd {tensorrtx}/yolov5/
4645
// update CLASS_NUM in yololayer.h if your model is trained on custom dataset
4746
mkdir build
4847
cd build
48+
cp {ultralytics}/yolov5/yolov5s.wts {tensorrtx}/yolov5/build
4949
cmake ..
5050
make
5151
sudo ./yolov5 -s [.wts] [.engine] [s/m/l/x/s6/m6/l6/x6 or c/c6 gd gw] // serialize model to plan file

0 commit comments

Comments
 (0)