Skip to content

Commit

Permalink
Merge pull request #802 from Samir-atra/patch-1
Browse files Browse the repository at this point in the history
Deleted some irrelevant comments
  • Loading branch information
OlafenwaMoses committed Jun 19, 2023
2 parents 64e6da4 + f34cf65 commit 339e48b
Showing 1 changed file with 1 addition and 26 deletions.
27 changes: 1 addition & 26 deletions scripts/pascal_voc_to_yolo.py
Expand Up @@ -51,7 +51,6 @@ def convert_annotation(input_ann_path):
)

return ann_list
# output_ann_path.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '\n')


def main(dataset_dir: str):
Expand Down Expand Up @@ -144,30 +143,6 @@ def main(dataset_dir: str):










# for dir_path in dirs:
# full_dir_path = cwd + '/' + dir_path
# output_path = full_dir_path +'/yolo/'

# if not os.path.exists(output_path):
# os.makedirs(output_path)

# image_paths = getImagesInDir(full_dir_path)
# list_file = open(full_dir_path + '.txt', 'w')

# for image_path in image_paths:
# list_file.write(image_path + '\n')
# convert_annotation(full_dir_path, output_path, image_path)
# list_file.close()

# print("Finished processing: " + dir_path)

if __name__ == "__main__":

parse = argparse.ArgumentParser(
Expand All @@ -179,4 +154,4 @@ def main(dataset_dir: str):
required=True,
)
args = parse.parse_args()
main(args.dataset_dir)
main(args.dataset_dir)

0 comments on commit 339e48b

Please sign in to comment.