Cli98 / DMNet

Official implementation for DMNet: Density map guided object detection in aerial image (CVPR 2020 EarthVision workshop)
https://openaccess.thecvf.com/content_CVPRW_2020/papers/w11/Li_Density_Map_Guided_Object_Detection_in_Aerial_Images_CVPRW_2020_paper.pdf
98 stars 15 forks source link

About generate density map? #3

Closed Mr-doraemon closed 3 years ago

Mr-doraemon commented 3 years ago

In _the Generate_densitty_mapofficial.py file, why does the coord array in the format_label function have 6 elements? And why do you mention 4coord+1class in the code.

 `if len(coord) != 6:`
       `# 4 coord + 1 class`
       `print("Failed to parse annotation!")`
       `exit()`

Looking forward to your reply.

Cli98 commented 3 years ago

In _the Generate_densitty_mapofficial.py file, why does the coord array in the format_label function have 6 elements? And why do you mention 4coord+1class in the code.

 `if len(coord) != 6:`
       `# 4 coord + 1 class`
       `print("Failed to parse annotation!")`
       `exit()`

Looking forward to your reply.

Hi @Mr-doraemon

The reason is, if you check the annotation for Visdrone 2018 dataset, first 6 coord are x,y,w,h,score,category

We only need x,y,w,h,category, that's 4 coord + 1 class.

The point at here is, just to make sure you load those parameters into the memory, that's enough.