Skip to content

Commit

Permalink
Update prepare_semantic3d_filelists.py
Browse files Browse the repository at this point in the history
  • Loading branch information
burui11087 authored Mar 28, 2019
1 parent 1216b1e commit 632ffd1
Showing 1 changed file with 3 additions and 14 deletions.
17 changes: 3 additions & 14 deletions data_conversions/prepare_semantic3d_filelists.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,24 +46,13 @@ def main():
filelist_i.write('../' + filename_h5)
for repeat_idx in range(args.repeat_num):
filelist.write('./filelists/train_files_g_%d.txt\n' % list_idx)

val_h5 = split_filelists['val']
random.shuffle(val_h5)
val_list = os.path.join(root, 'val_data_files.txt')
print('{}-Saving {}...'.format(datetime.now(), val_list))
with open(val_list, 'w') as filelist:
list_num = math.ceil(len(val_h5) / args.h5_num)
for list_idx in range(list_num):
val_list_i = os.path.join(root, 'filelists', 'val_files_g_%d.txt' % list_idx)
with open(val_list_i, 'w') as filelist_i:
for h5_idx in range(args.h5_num):
filename_idx = list_idx * args.h5_num + h5_idx
if filename_idx > len(val_h5) - 1:
break
filename_h5 = val_h5[filename_idx]
filelist_i.write('../' + filename_h5)
for repeat_idx in range(args.repeat_num):
filelist.write('./filelists/val_files_g_%d.txt\n' % list_idx)
for filename_h5 in val_h5:
filelist.write(filename_h5)

test_h5 = split_filelists['test']
test_list = os.path.join(root, 'test_files.txt')
Expand Down

0 comments on commit 632ffd1

Please sign in to comment.