diff --git a/genCarPlate.py b/genCarPlate.py index e3020c8..47f7ae1 100644 --- a/genCarPlate.py +++ b/genCarPlate.py @@ -72,7 +72,7 @@ def gen_batch(self, perSize, outDir): for i in range(perSize*31-1): outputPath = outDir + str(i // perSize) + "/" if (not os.path.exists(outputPath)): - os.mkdir(outputPath) + os.makedirs(outputPath) plate_str = self.gen_plate_string(i, perSize) img, loc = self.generate(plate_str) if img is None: @@ -95,4 +95,4 @@ def main(args): G.gen_batch(args.gen_count_per_province, args.out_dir) if __name__ == '__main__': - main(parse_args()) \ No newline at end of file + main(parse_args())