Skip to content

Commit

Permalink
modify reamde
Browse files Browse the repository at this point in the history
  • Loading branch information
mpj1234 committed Jul 12, 2024
1 parent 91f9447 commit dba8c4a
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions yolov5/yolov5_trt10/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ cd [PATH-TO-TENSORRTX]/yolov5/yolov5_trt10
# Update kNumClass in src/config.h if your model is trained on custom dataset
mkdir build
cd build
cp [PATH-TO-ultralytics-yolov5]/yolov5sn-cls.wts .
cp [PATH-TO-ultralytics-yolov5]/yolov5sn-cls.wts .
cmake ..
make

Expand All @@ -64,7 +64,7 @@ wget https://github.com/joannzhang00/ImageNet-dataset-classes-labels/blob/main/i
// Ensure the yolov5n-cls.engine
python yolov5_cls_trt.py
# faq: in windows bug pycuda._driver.LogicError
# faq: in linux bug Segmentation fault
# faq: in linux bug Segmentation fault
# Add the following code to the py file:
# import pycuda.autoinit
# import pycuda.driver as cuda
Expand All @@ -77,7 +77,7 @@ cd [PATH-TO-TENSORRTX]/yolov5/yolov5_trt10
# Update kNumClass in src/config.h if your model is trained on custom dataset
mkdir build
cd build
cp [PATH-TO-ultralytics-yolov5]/yolov5n.wts .
cp [PATH-TO-ultralytics-yolov5]/yolov5n.wts .
cmake ..
make

Expand All @@ -96,4 +96,4 @@ make
4. serialize the model and test

## More Information
See the readme in [home page.](https://github.com/wang-xinyu/tensorrtx)
See the readme in [home page.](https://github.com/wang-xinyu/tensorrtx)

0 comments on commit dba8c4a

Please sign in to comment.