13 lines
291 B
Python
13 lines
291 B
Python
|
|
|
|
def get_dataset_info(dataset_type):
|
|
"""define dataset_name and its dataroot"""
|
|
root = ''
|
|
if dataset_type == 'LEVIR_CD':
|
|
root = 'path-to-LEVIR_CD-dataroot'
|
|
# add more dataset ...
|
|
else:
|
|
raise TypeError("not define the %s" % dataset_type)
|
|
|
|
return root
|