Skip to content

Commit

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

**1. Prepare custom dataset files**

1.1 Make sure the labels format is [poly classname diffcult], e.g.,
1.1 Make sure the labels format is [poly classname diffcult], e.g., You can set **diffcult=0**
```
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**)**
![image](https://user-images.githubusercontent.com/72599120/159213229-b7c2fc5c-b140-4f10-9af8-2cbc405b0cd3.png)


Expand All @@ -25,7 +24,6 @@ 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)**

1.3 Make sure your dataset structure same as:
```
Expand All @@ -48,7 +46,8 @@ parent
```

**Note:**
* DOTA is a high resolution image dataset, so it must be splited before training to get a better performance.
* DOTA is a high resolution image dataset, so it needs to be splited before training/testing to get better performance.

**2. Train**

2.1 Train with specified GPUs. (for example with GPU=3)
Expand Down

0 comments on commit f6390db

Please sign in to comment.