update readme

aqntks · Feb 22, 2022 · 26aae38 · 26aae38
1 parent 08c66af
commit 26aae38
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -45,7 +45,7 @@ $ pip install -r requirements.txt
 ## OCR
 
 ```bash
-$ python main.py --gpu 0 --lang en ko
+$ python main.py --gpu 0 --lang en/ko
 $ python main.py --gpu 0 --lang en
 $ python main.py --gpu -1 --lang ko         # --gpu -1 : cpu mode
 ```
@@ -147,10 +147,10 @@ names: ['title', 'name', 'personal_id', 'text_box_1', 'text_box_2', 'price', 'ad
 ## Train Detection Model
 
 ```bash
-$ python train.py --data data/custom.yaml --weights 'yolov5s.pt' --img 640 --batch-size 64 --epochs 300
-                                                     yolov5m.pt        960              40          100
-                                                     yolov5l.pt        480              24           50 
-                                                     yolov5x.pt        320              16           30 
+$ python train.py --data data/custom.yaml --weights yolov5s.pt --img 640 --batch-size 64 --epochs 300
+                                                     yolov5m.pt      960              40          100
+                                                     yolov5l.pt      480              24           50 
+                                                     yolov5x.pt      320              16           30 
 ```
 
 ## Setting Config

diff --git a/README_en.md b/README_en.md
@@ -44,7 +44,7 @@ $ pip install -r requirements.txt
 ## OCR
 
 ```bash
-$ python main.py --gpu 0 --lang en ko
+$ python main.py --gpu 0 --lang en/ko
 $ python main.py --gpu 0 --lang en
 $ python main.py --gpu -1 --lang ko         # --gpu -1 : cpu mode
 ```
@@ -145,10 +145,10 @@ names: ['title', 'name', 'personal_id', 'text_box_1', 'text_box_2', 'price', 'ad
 ## Train Detection Model
 
 ```bash
-$ python train.py --data data/custom.yaml --weights 'yolov5s.pt' --img 640 --batch-size 64 --epochs 300
-                                                     yolov5m.pt        960              40          100
-                                                     yolov5l.pt        480              24           50 
-                                                     yolov5x.pt        320              16           30 
+$ python train.py --data data/custom.yaml --weights yolov5s.pt --img 640 --batch-size 64 --epochs 300
+                                                    yolov5m.pt       960              40          100
+                                                    yolov5l.pt       480              24           50 
+                                                    yolov5x.pt       320              16           30 
 ```
 
 ## Setting Config