search

VSainteuf / pytorch-psetae

PyTorch implementation of the model presented in "Satellite Image Time Series Classification with Pixel-Set Encoders and Temporal Self-Attention"
MIT License
178 stars 38 forks source link

Possibility on training binary classes? #25

Open manapshymyr-OB opened 2 months ago

manapshymyr-OB commented 2 months ago

Hi @VSainteuf , I am curious if it is possible to find specific crop types based on the your scripts? For example, I might be interested only in rice, or cannabis detection... If yes, could you please guide on it?

VSainteuf commented 2 months ago

Hi @manapshymyr-OB, I pasted here the class names of each label in the 44class label set of the dataset. You can then look for samples of the class you are interested in.

Label Label name
0 Undefined
1 Meadow
2 Spring Durum Wheat
3 Winter Durum Wheat
4 Spring Cereal
5 Summer Cereal
6 Winter Cereal
7 Corn and Zea
8 Sorghum / Millet / Moha
9 Cereal
10 Buckwheat
11 Hemp
12 Leguminous Fodder
13 Other Fodder
14 Winter Rapeseed
15 Spring Rapeseed
16 Sunflower
17 Oleaginous
18 Soy
19 Protein Crop
20 Rice
21 Beetroot
22 Flax
23 Potato
24 Tobacco
25 Banana
26 Coffee / Cocoa
27 Dry Vegetables
28 Fruits, Vegetables, Flowers
29 Lavender
30 Fragrant Plants, Aromatic and Medicinal
31 Ligneous
32 Olive groves
33 Orchards
34 Grapevine
35 Short Rotation Coppice
36 Non-agricultural Surface
37 Built Area
38 Biomass
39 Wood
40 Sugar cane
41 Water, Marsh
42 Hop
43 Other