Spaces:
Build error
Build error
| # 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', | |
| }, | |
| } | |