Skip to content

Commit 9dfb126

Browse files
nilslehnilsleh
authored
Add SODA-A dataset (microsoft#2575)
* soda * plotting * legend loc * tests * tests * import skip * requests * requests * tests * requests * comment * rm import skip * ruff * requests * boxes
1 parent 6067656 commit 9dfb126

File tree

23 files changed

+1433
-0
lines changed

23 files changed

+1433
-0
lines changed

‎docs/api/datasets.rst

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -482,6 +482,12 @@ So2Sat
482482

483483
.. autoclass:: So2Sat
484484

485+
486+
SODA
487+
^^^^
488+
489+
.. autoclass:: SODAA
490+
485491
SpaceNet
486492
^^^^^^^^
487493

‎docs/api/datasets/non_geo_datasets.csv

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ Dataset,Task,Source,License,# Samples,# Classes,Size (px),Resolution (m),Bands
5353
`SKIPP'D`_,R,"Fish-eye","CC-BY-4.0","363,375",-,64x64,-,RGB
5454
`SkyScript`_,IC,"NAIP, orthophotos, Planet SkySat, Sentinel-2, Landsat 8--9",MIT,5.2M,-,100--1000,0.1--30,RGB
5555
`So2Sat`_,C,Sentinel-1/2,"CC-BY-4.0","400,673",17,32x32,10,"SAR, MSI"
56+
`SODA`_,OD,Aerial,"CC-BY-NC-4.0","2513",9,"~2700x~4800","RGB"
5657
`SpaceNet`_,I,WorldView-2/3 Planet Lab Dove,"CC-BY-SA-4.0","1,889--28,728",2,102--900,0.5--4,MSI
5758
`SSL4EO`_-L,T,Landsat,"CC0-1.0",1M,-,264x264,30,MSI
5859
`SSL4EO`_-S12,T,Sentinel-1/2,"CC-BY-4.0",1M,-,264x264,10,"SAR, MSI"

‎tests/data/soda/Annotations.zip

3.87 KB
Binary file not shown.

‎tests/data/soda/Annotations/test/00006.json

Lines changed: 117 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,117 @@
1+
{
2+
"type": "instance",
3+
"images": {
4+
"file_name": "00006.jpg",
5+
"height": 32,
6+
"width": 32,
7+
"id": 6
8+
},
9+
"annotations": [
10+
{
11+
"poly": [
12+
15.0,
13+
11.0,
14+
23.0,
15+
11.0,
16+
23.0,
17+
19.0,
18+
15.0,
19+
19.0
20+
],
21+
"area": 1724.0,
22+
"category_id": 4,
23+
"image_id": 6,
24+
"id": 1
25+
},
26+
{
27+
"poly": [
28+
21.0,
29+
6.5,
30+
29.0,
31+
6.5,
32+
29.0,
33+
11.5,
34+
21.0,
35+
11.5
36+
],
37+
"area": 1607.0,
38+
"category_id": 6,
39+
"image_id": 6,
40+
"id": 2
41+
},
42+
{
43+
"poly": [
44+
21.0,
45+
16.0,
46+
27.0,
47+
16.0,
48+
27.0,
49+
22.0,
50+
21.0,
51+
22.0
52+
],
53+
"area": 1632.0,
54+
"category_id": 5,
55+
"image_id": 6,
56+
"id": 3
57+
},
58+
{
59+
"poly": [
60+
8.5,
61+
5.0,
62+
13.5,
63+
5.0,
64+
13.5,
65+
11.0,
66+
8.5,
67+
11.0
68+
],
69+
"area": 533.0,
70+
"category_id": 8,
71+
"image_id": 6,
72+
"id": 4
73+
}
74+
],
75+
"categories": [
76+
{
77+
"id": 0,
78+
"name": "airplane"
79+
},
80+
{
81+
"id": 1,
82+
"name": "helicopter"
83+
},
84+
{
85+
"id": 2,
86+
"name": "small-vehicle"
87+
},
88+
{
89+
"id": 3,
90+
"name": "large-vehicle"
91+
},
92+
{
93+
"id": 4,
94+
"name": "ship"
95+
},
96+
{
97+
"id": 5,
98+
"name": "container"
99+
},
100+
{
101+
"id": 6,
102+
"name": "storage-tank"
103+
},
104+
{
105+
"id": 7,
106+
"name": "swimming-pool"
107+
},
108+
{
109+
"id": 8,
110+
"name": "windmill"
111+
},
112+
{
113+
"id": 9,
114+
"name": "ignore"
115+
}
116+
]
117+
}

‎tests/data/soda/Annotations/test/00007.json

Lines changed: 85 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,85 @@
1+
{
2+
"type": "instance",
3+
"images": {
4+
"file_name": "00007.jpg",
5+
"height": 32,
6+
"width": 32,
7+
"id": 7
8+
},
9+
"annotations": [
10+
{
11+
"poly": [
12+
21.5,
13+
7.0,
14+
30.5,
15+
7.0,
16+
30.5,
17+
13.0,
18+
21.5,
19+
13.0
20+
],
21+
"area": 435.0,
22+
"category_id": 3,
23+
"image_id": 7,
24+
"id": 1
25+
},
26+
{
27+
"poly": [
28+
5.5,
29+
19.5,
30+
14.5,
31+
19.5,
32+
14.5,
33+
26.5,
34+
5.5,
35+
26.5
36+
],
37+
"area": 1770.0,
38+
"category_id": 6,
39+
"image_id": 7,
40+
"id": 2
41+
}
42+
],
43+
"categories": [
44+
{
45+
"id": 0,
46+
"name": "airplane"
47+
},
48+
{
49+
"id": 1,
50+
"name": "helicopter"
51+
},
52+
{
53+
"id": 2,
54+
"name": "small-vehicle"
55+
},
56+
{
57+
"id": 3,
58+
"name": "large-vehicle"
59+
},
60+
{
61+
"id": 4,
62+
"name": "ship"
63+
},
64+
{
65+
"id": 5,
66+
"name": "container"
67+
},
68+
{
69+
"id": 6,
70+
"name": "storage-tank"
71+
},
72+
{
73+
"id": 7,
74+
"name": "swimming-pool"
75+
},
76+
{
77+
"id": 8,
78+
"name": "windmill"
79+
},
80+
{
81+
"id": 9,
82+
"name": "ignore"
83+
}
84+
]
85+
}

‎tests/data/soda/Annotations/train/00001.json

Lines changed: 117 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,117 @@
1+
{
2+
"type": "instance",
3+
"images": {
4+
"file_name": "00001.jpg",
5+
"height": 32,
6+
"width": 32,
7+
"id": 1
8+
},
9+
"annotations": [
10+
{
11+
"poly": [
12+
12.0,
13+
17.0,
14+
18.0,
15+
17.0,
16+
18.0,
17+
25.0,
18+
12.0,
19+
25.0
20+
],
21+
"area": 1929.0,
22+
"category_id": 6,
23+
"image_id": 1,
24+
"id": 1
25+
},
26+
{
27+
"poly": [
28+
13.0,
29+
18.5,
30+
19.0,
31+
18.5,
32+
19.0,
33+
25.5,
34+
13.0,
35+
25.5
36+
],
37+
"area": 1680.0,
38+
"category_id": 1,
39+
"image_id": 1,
40+
"id": 2
41+
},
42+
{
43+
"poly": [
44+
17.0,
45+
17.5,
46+
25.0,
47+
17.5,
48+
25.0,
49+
24.5,
50+
17.0,
51+
24.5
52+
],
53+
"area": 423.0,
54+
"category_id": 7,
55+
"image_id": 1,
56+
"id": 3
57+
},
58+
{
59+
"poly": [
60+
21.5,
61+
5.0,
62+
28.5,
63+
5.0,
64+
28.5,
65+
11.0,
66+
21.5,
67+
11.0
68+
],
69+
"area": 983.0,
70+
"category_id": 4,
71+
"image_id": 1,
72+
"id": 4
73+
}
74+
],
75+
"categories": [
76+
{
77+
"id": 0,
78+
"name": "airplane"
79+
},
80+
{
81+
"id": 1,
82+
"name": "helicopter"
83+
},
84+
{
85+
"id": 2,
86+
"name": "small-vehicle"
87+
},
88+
{
89+
"id": 3,
90+
"name": "large-vehicle"
91+
},
92+
{
93+
"id": 4,
94+
"name": "ship"
95+
},
96+
{
97+
"id": 5,
98+
"name": "container"
99+
},
100+
{
101+
"id": 6,
102+
"name": "storage-tank"
103+
},
104+
{
105+
"id": 7,
106+
"name": "swimming-pool"
107+
},
108+
{
109+
"id": 8,
110+
"name": "windmill"
111+
},
112+
{
113+
"id": 9,
114+
"name": "ignore"
115+
}
116+
]
117+
}

0 commit comments

Comments
 (0)