6D prediction files now working

This commit is contained in:
Si11ium
2020-06-26 08:33:59 +02:00
parent 2a7a236b89
commit 358d692699
3 changed files with 5 additions and 5 deletions

View File

@ -26,8 +26,8 @@ main_arg_parser.add_argument("--data_root", type=str, default='data', help="")
main_arg_parser.add_argument("--data_refresh", type=strtobool, default=False, help="")
main_arg_parser.add_argument("--data_dataset_type", type=str, default='ShapeNetPartSegDataset', help="")
main_arg_parser.add_argument("--data_cluster_type", type=str, default='grid', help="")
main_arg_parser.add_argument("--data_normals_as_cords", type=strtobool, default=False, help="")
main_arg_parser.add_argument("--data_poly_as_plane", type=strtobool, default=True, help="")
main_arg_parser.add_argument("--data_normals_as_cords", type=strtobool, default=True, help="")
main_arg_parser.add_argument("--data_poly_as_plane", type=strtobool, default=False, help="")
# Transformations
# main_arg_parser.add_argument("--transformations_to_tensor", type=strtobool, default=False, help="")