You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
### Give a ⭐️ if this project helped you. If you use it, please consider citing:
6
-
```arxiv
7
-
article{huang2021general,
8
-
title = {A General Gaussian Heatmap Labeling for Arbitrary-Oriented Object Detection},
9
-
author = {Huang, Zhanchao and Li, Wei and Xia, Xiang-Gen and Tao, Ran},
10
-
year = {2021},
11
-
journal = {arXiv preprint arXiv:2109.12848},
12
-
eprint = {2109.12848},
13
-
eprinttype = {arxiv},
14
-
archiveprefix = {arXiv}
15
-
}
16
-
```
17
-
18
-
### 👹 Abstract of the paper
5
+
### Give a ⭐️ if this project helped you. If you use it, please consider citing:
6
+
```arxiv
7
+
article{huang2021general,
8
+
title = {A General Gaussian Heatmap Labeling for Arbitrary-Oriented Object Detection},
9
+
author = {Huang, Zhanchao and Li, Wei and Xia, Xiang-Gen and Tao, Ran},
10
+
year = {2021},
11
+
journal = {arXiv preprint arXiv:2109.12848},
12
+
eprint = {2109.12848},
13
+
eprinttype = {arxiv},
14
+
archiveprefix = {arXiv}
15
+
}
16
+
```
17
+
18
+
### 👹 Abstract of the paper
19
19
20
20
Recently, many arbitrary-oriented object detection (AOOD) methods have been proposed and attracted widespread attention in many fields. However, most of them are based on anchor-boxes or standard Gaussian heatmaps. Such label assignment strategy may not only fail to reflect the shape and direction characteristics of arbitrary-oriented objects, but also have high parameter-tuning efforts. In this paper, a novel AOOD method called General Gaussian Heatmap Labeling (GGHL) is proposed. Specifically, an anchor-free object adaptation label assignment (OLA) strategy is presented to define the positive candidates based on two-dimensional (2-D) oriented Gaussian heatmaps, which reflect the shape and direction features of arbitrary-oriented objects. Based on OLA, an oriented-boundingbox (OBB) representation component (ORC) is developed to indicate OBBs and adjust the Gaussian center prior weights to fit the characteristics of different objects adaptively through neural network learning. Moreover, a joint-optimization loss (JOL) with area normalization and dynamic confidence weighting is designed to refine the misalign optimal results of different subtasks. Extensive experiments on public datasets demonstrate that the proposed GGHL improves the AOOD performance with low parameter-tuning and time costs. Furthermore, it is generally applicable to most AOOD methods to improve their performance including lightweight models on embedded platforms.
#### 🐣 🐤 🐥 11.9: The model weight has been released. You can download it and put it in the ./weight folder, and then modify the weight path in test.py to test and get the results reported in the paper. The download link is given in the introduction later.
38
+
*#### 🐣 🐤 🐥 11.9: The model weight has been released. You can download it and put it in the ./weight folder, and then modify the weight path in test.py to test and get the results reported in the paper. The download link is given in the introduction later.
39
39
论文结果对应的模型权重可以下载了(终于发工资把网盘续上了~)
40
40
41
-
#### 🐞 11.8:I plan to write a tutorial on data preprocessing and explanation of algorithms and codes, which is expected to be launched in December
41
+
*#### 🐞 11.8:I plan to write a tutorial on data preprocessing and explanation of algorithms and codes, which is expected to be launched in December
42
42
打算写一个数据预处理的教程和算法、代码的讲解,预计12月上线
43
43
44
-
#### 🦄 11.7: All updates of GGHL have been completed. Welcome to use it. If you have any questions, you can leave a message at the issue. Thank you.
44
+
*#### 🦄 11.7: All updates of GGHL have been completed. Welcome to use it. If you have any questions, you can leave a message at the issue. Thank you.
45
45
1.0版本全部更新完成了,欢迎使用,有任何问题可以在issue留言,谢谢。接下来会不断更新和完善
46
46
47
47
@@ -56,7 +56,8 @@ Second, install the dependent libraries in [requirements.txt](https://github.com
1.[DOTA dataset](https://captain-whu.github.io/DOTA/dataset.html) and its [devkit](https://github.com/CAPTAIN-WHU/DOTA_devkit)
71
73
72
-
(1) VOC Format
74
+
#### (1) VOC Format
73
75
You need to write a script to convert them into the train.txt file required by this repository and put them in the ./dataR folder.
74
76
For the specific format of the train.txt file, see the example in the /dataR folder.
75
77
@@ -97,7 +99,8 @@ sh train_GGHL_dist.sh
97
99
```python
98
100
python test.py
99
101
```
100
-
102
+
103
+
101
104
## ☃️❄️ 5.Weights
102
105
The pre-trained weights and trained models are available from [Google Drive](https://drive.google.com/file/d/13yrGQTcA3xLf6TPsAA1cVTF0rAUk6Keg/view?usp=sharing) or [Baidu Disk](https://pan.baidu.com/s/1aZ-bnNUAqJHqfOThK4tm5A) (password: 2dm8)
0 commit comments