Skip to content

Commit

Permalink
Update GetStart.md
Browse files Browse the repository at this point in the history
  • Loading branch information
hukaixuan19970627 authored Jan 9, 2022
1 parent f265b60 commit 0a763ae
Showing 1 changed file with 13 additions and 7 deletions.
20 changes: 13 additions & 7 deletions docs/GetStart.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,14 +4,18 @@ This page provides basic usage about yolov5-obb. For installation instructions,

# Train a model

1. **Prepare custom dataset files**
1.1 Make sure the labels format is [poly classname diffcult], e.g., (*Note: You can set diffcult=0)
**1. Prepare custom dataset files**

1.1 Make sure the labels format is [poly classname diffcult], e.g.,
```
x1 y1 x2 y2 x3 y3 x4 y4 classname diffcult
1686.0 1517.0 1695.0 1511.0 1711.0 1535.0 1700.0 1541.0 large-vehicle 1
x1 y1 x2 y2 x3 y3 x4 y4 classname diffcult
1686.0 1517.0 1695.0 1511.0 1711.0 1535.0 1700.0 1541.0 large-vehicle 1
```
**(*Note: You can set diffcult=0)**

1.2 Split the dataset. (*Note: High resolution image dataset needs to be splited to get better performance in small objects)

1.2 Split the dataset.
```shell
cd yolov5_obb
python DOTA_devkit/ImgSplit_multi_process.py
Expand All @@ -20,8 +24,10 @@ or Use the orignal dataset.
```shell
cd yolov5_obb
```
**(*Note: High resolution image dataset needs to be splited to get better performance in small objects)**


2. **Train**
**2. Train**
2.1 Train with specified GPUs. (for example with GPU=3)

```shell
Expand Down Expand Up @@ -139,4 +145,4 @@ python detect.py --weights 'runs/train/yolov5m_finetune/weights/best.pt' \
--img 2048 --device 2 --hide-labels --hide-conf
```

*If you want to evaluate the result on DOTA test-dev, please zip the poly format results files and submit it to the [evaluation server](https://captain-whu.github.io/DOTA/index.html).
***If you want to evaluate the result on DOTA test-dev, please zip the poly format results files and submit it to the [evaluation server](https://captain-whu.github.io/DOTA/index.html).**

0 comments on commit 0a763ae

Please sign in to comment.