Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
sivaji123256 authored Aug 23, 2022
1 parent a92b59f commit 7732d75
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
# YoloV6
YoloV6 Custom Dataset Training
## 1. Clone the repo as follows:
git clone https://github.com/sivaji123256/YoloV6.git
!git clone https://github.com/sivaji123256/YoloV6.git
## 2. Change the directory yolov6 parent folder
* Run the following command to install required packages
!pip install -r requirements.txt
!pip install -r requirements.txt
* In the parent folder access dataset.yaml file and add the required path to the train dataset and test dateset and change other parametrs as required
* In the weights add required yolov6 pretrained weights.pt files
## 3. Training on custom dataset
Expand All @@ -13,6 +13,6 @@ YoloV6 Custom Dataset Training
* change the parameters as required
## 4. For evaluation
* Run the following command
!python tools/eval.py --data dataset.yaml --weights runs/train/exp/weights/best_ckpt.pt --device 0
!python tools/eval.py --data dataset.yaml --weights runs/train/exp/weights/best_ckpt.pt --device 0
## 5. For inference run the following command
!python tools/infer.py --weights runs/train/exp/weights/best_ckpt.pt --source 100.jpg --yaml dataset.yaml --device 0
!python tools/infer.py --weights runs/train/exp/weights/best_ckpt.pt --source 100.jpg --yaml dataset.yaml --device 0

0 comments on commit 7732d75

Please sign in to comment.