EarthEmbeddingExplorer / models /FarSLIP /open_clip_train /benchmark_dataset_info.py
ML4RS-Anonymous's picture
Upload all files
eb1aec4 verified
# Please replace this with the root directory of benchmark datasets
# BENCHMARK_DATASET_ROOT_DIR = 'BENCHMARK_DATASET_ROOT_DIR/'
from open_clip_train.config import BENCHMARK_DATASET_ROOT_DIR
BENCHMARK_DATASET_INFOMATION = {
'aid': {
'classification_mode': 'multiclass',
'test_data': BENCHMARK_DATASET_ROOT_DIR + '/aid/aid_img_txt_pairs_test.csv',
'classnames': BENCHMARK_DATASET_ROOT_DIR + '/aid/classnames.txt',
'csv_separator': ',',
'csv_img_key': 'filepath',
'csv_class_key': 'label',
},
'eurosat': {
'classification_mode': 'multiclass',
'test_data': BENCHMARK_DATASET_ROOT_DIR + '/eurosat/eurosat_img_txt_pairs_test.csv',
'classnames': BENCHMARK_DATASET_ROOT_DIR + '/eurosat/classnames.txt',
'csv_separator': ',',
'csv_img_key': 'filepath',
'csv_class_key': 'label',
},
'fmow': {
'classification_mode': 'multiclass',
'test_data': BENCHMARK_DATASET_ROOT_DIR + '/fmow/fmow_img_txt_pairs_val.csv',
'classnames': BENCHMARK_DATASET_ROOT_DIR + '/fmow/classnames.txt',
'csv_separator': ',',
'csv_img_key': 'filepath',
'csv_class_key': 'label',
},
'nwpu': {
'classification_mode': 'multiclass',
'test_data': BENCHMARK_DATASET_ROOT_DIR + '/nwpu/img_txt_pairs_train.csv',
'classnames': BENCHMARK_DATASET_ROOT_DIR + '/nwpu/classnames.txt',
'csv_separator': ',',
'csv_img_key': 'filepath',
'csv_class_key': 'label',
},
'patternnet': {
'classification_mode': 'multiclass',
'test_data': BENCHMARK_DATASET_ROOT_DIR + '/patternnet/img_txt_pairs_train.csv',
'classnames': BENCHMARK_DATASET_ROOT_DIR + '/patternnet/classnames.txt',
'csv_separator': ',',
'csv_img_key': 'filepath',
'csv_class_key': 'label',
},
'SkyScript_cls': {
'classification_mode': 'multiclass',
'test_data': BENCHMARK_DATASET_ROOT_DIR + '/SkyScript_cls/img_txt_pairs_val.csv',
'classnames': BENCHMARK_DATASET_ROOT_DIR + '/SkyScript_cls/classnames.txt',
'csv_separator': ',',
'csv_img_key': 'filepath',
'csv_class_key': 'label',
},
'millionaid': {
'classification_mode': 'multiclass',
'test_data': BENCHMARK_DATASET_ROOT_DIR + '/millionaid/img_txt_pairs_train.csv',
'classnames': BENCHMARK_DATASET_ROOT_DIR + '/millionaid/classnames.txt',
'csv_separator': ',',
'csv_img_key': 'filepath',
'csv_class_key': 'label',
},
'rsicb': {
'classification_mode': 'multiclass',
'test_data': BENCHMARK_DATASET_ROOT_DIR + '/rsicb256/img_txt_pairs_train.csv',
'classnames': BENCHMARK_DATASET_ROOT_DIR + '/rsicb256/classnames.txt',
'csv_separator': ',',
'csv_img_key': 'filepath',
'csv_class_key': 'label',
},
}