Skip to content

Commit 7d6dfaf

Browse files
authored
Update README.md
1 parent 17d124f commit 7d6dfaf

File tree

1 file changed

+28
-1
lines changed

1 file changed

+28
-1
lines changed

README.md

+28-1
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,29 @@
1-
# labeling_images_with_trained_yolo_weight
1+
# Labeling Images with Trained Yolo Weight
22

3+
This function I wrote performs the image and txt saving process. You can also use this function on a model you have trained before. If your model works well enough, it will easily detect the images and save the labels. You can edit the saved data and train with your new data quickly.
4+
5+
### How to use
6+
first of all, tensorrt transform using this repository of the yolo weights you have trained
7+
~~~~~
8+
https://github.com/jkjung-avt/tensorrt_demos
9+
~~~~~~~~~~~
10+
You should edit `yolo/obj.names` according to your project class names
11+
12+
And You should change `plugins` and `utils` your tensorrt_demos file
13+
14+
### Run the application
15+
16+
The input parameters can be changed using the command line :
17+
~~~~~
18+
python3 img_test.py -i <input_images_file> -w <trt_weight_name> -t <image_type(jpg,png...)>
19+
~~~~~~~~~~~~
20+
21+
For running with custom weights :
22+
~~~~
23+
python3 img_test.py -i images -w yolov4-tiny -t jpg
24+
~~~~~~~~~
25+
26+
To display the help :
27+
~~~~
28+
python3 img_test.py -h
29+
~~~~~~

0 commit comments

Comments
 (0)