91bb9f05e95d0902f5fcd9c4ef8dee5f193c8bf7,create_training_dataset.py,,,#,183

Before Change

        download_coco_data(args.coco_image_dir)

    image_directory = os.path.join(args.coco_image_dir, "train2014/")
    if not os.path.isdir(image_directory):
        sys.exit(
            "Error: There is no `train2014/` folder in the provided directory."
        )

    CocoPreprocessor.process_images(
        image_directory,
        args.output,
        image_size=(args.img_height, args.img_width),

After Change

        download_coco_data(image_directory)
        image_directory = os.path.join(image_directory, "train2014")

    image_directory = os.path.join(args.image_dir)
    DatasetCreator.process_images(
        image_directory,
        args.output,
        image_size=(args.img_height, args.img_width),

Instances