GitLab is experiencing storage issues, linked with the Windows servers shutdown. Several features are affected by this disruption.

Check the info page for updates

Commit aff7a83a authored by gencersumbul's avatar gencersumbul

Update prep_splits_BigEarthNet-19.py

parent 84056441
......@@ -7,8 +7,8 @@
#
# prep_splits_BigEarthNet-19.py --help can be used to learn how to use this script.
#
# Author: Gencer Sumbul, http://www.user.tu-berlin.de/gencersumbul/, Jian Kang, https://www.rsim.tu-berlin.de/menue/team/dring_jian_kang/
# Email: gencer.suembuel@tu-berlin.de, jian.kang@tu-berlin.de
# Author: Gencer Sumbul, http://www.user.tu-berlin.de/gencersumbul/
# Email: gencer.suembuel@tu-berlin.de
# Date: 16 Jan 2020
# Version: 1.0.1
# Usage: prep_splits_BigEarthNet-19.py [-h] [-r ROOT_FOLDER] [-o OUT_FOLDER] [--update_json]
......@@ -20,7 +20,6 @@ import os
import csv
import json
from tensorflow_utils import prep_tf_record_files
from pytorch_utils import prep_lmdb_files
GDAL_EXISTED = False
RASTERIO_EXISTED = False
......@@ -40,7 +39,7 @@ if __name__ == "__main__":
'flag for adding BigEarthNet-19 labels to the json file of each patch')
parser.add_argument('-n', '--splits', dest = 'splits', help =
'csv files each of which contain list of patch names, patches with snow, clouds, and shadows already excluded', nargs = '+')
parser.add_argument('-l', '--library', type=str, dest = 'library', help="Limit search to Sentinel mission", choices=['tensorflow', 'pytorch'])
parser.add_argument('-l', '--library', type=str, dest = 'library', help="Limit search to Sentinel mission", choices=['tensorflow'])
args = parser.parse_args()
......@@ -110,14 +109,3 @@ if __name__ == "__main__":
RASTERIO_EXISTED,
UPDATE_JSON
)
elif args.library == 'pytorch':
prep_lmdb_files(
args.root_folder,
args.out_folder,
patch_names_list,
label_indices,
GDAL_EXISTED,
RASTERIO_EXISTED,
UPDATE_JSON
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment