Source code for locpix.scripts.development.file_names

#!/usr/bin/env python
"""Get file names without extensions from
"""
import os
import argparse


[docs] def main(): parser = argparse.ArgumentParser(description="Get file names") parser.add_argument( "-i", "--project_directory", action="store", type=str, help="the location of the project directory", required=True, ) parser.add_argument( "-e", "--extension", action="store", type=str, help="extension to remove", required=True, ) args = parser.parse_args() folder = os.path.join(args.project_directory, "preprocess/no_gt_label") files = os.listdir(folder) for file in files: print(file.removesuffix(args.extension))
if __name__ == "__main__": main()