|
@@ -27,7 +27,7 @@ def main():
|
|
)
|
|
)
|
|
parser.add_argument('input_dir', help='input annotated directory')
|
|
parser.add_argument('input_dir', help='input annotated directory')
|
|
parser.add_argument('output_dir', help='output dataset directory')
|
|
parser.add_argument('output_dir', help='output dataset directory')
|
|
- parser.add_argument('--labels', help='labels file')
|
|
|
|
|
|
+ parser.add_argument('--labels', help='labels file', required=True)
|
|
args = parser.parse_args()
|
|
args = parser.parse_args()
|
|
|
|
|
|
if osp.exists(args.output_dir):
|
|
if osp.exists(args.output_dir):
|