How to create a folder in python?

Question

I am running a code in python where I get images from input file, and create another folder as output and a file csv. The code that I run is as below:

# import the necessary packages
from PIL import Image
import argparse
import random
import shutil
import glob2
import uuid

# construct the argument parse and parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-i", "--input", required = True,
    help = "input directory of images")
ap.add_argument("-o", "--output", required = True,
    help = "output directory")
ap.add_argument("-c", "--csv", required = True,
    help = "path to CSV file for image counts")
args = vars(ap.parse_args())

# open the output file for writing
output = open(args["csv"], "w")

# loop over the input images
for imagePath in glob2.iglob(args["input"] + "/*/*.jpg"):
    # generate a random filename for the image and copy it to
    # the output location
    filename = str(uuid.uuid4()) + ".jpg"
    shutil.copy(imagePath, args["output"] + "/" + filename)

    # there is a 1 in 500 chance that multiple copies of this
    # image will be used
    if random.randint(0, 500) == 0:
        # initialize the number of times the image is being
        # duplicated and write it to the output CSV file
        numTimes = random.randint(1, 8)
        output.write("%s,%d\n" % (filename, numTimes))

        # loop over a random number of times for this image to
        # be duplicated
        for i in range(0, numTimes):
            image = Image.open(imagePath)

            # randomly resize the image, perserving aspect ratio
            factor = random.uniform(0.95, 1.05)
            width = int(image.size[0] * factor)
            ratio = width / float(image.size[0])
            height = int(image.size[1] * ratio)
            image = image.resize((width, height), Image.ANTIALIAS)

            # generate a random filename for the image and copy
            # it to the output directory
            adjFilename = str(uuid.uuid4()) + ".jpg"
            shutil.copy(imagePath, args["output"] + "/" + adjFilename)

# close the output file
output.close()

After running the code I get only csv file, but I don't get output folder. The way I run the code is:

python gather.py --input 101_ObjectCategories --output images --csv output.csv

Please can you help me how to solve the problem, because I need the output folder for next steps, running next functions.

The error I get Traceback (most recent call last): File "...\code\gather.py", line 31, in <module> shutil.copy(imagePath, args["output"] + "/" + filename) File "...\Anaconda3\lib\shutil.py", line 248, in copy copyfile(src, dst, follow_symlinks=follow_symlinks) File "...\Anaconda3\lib\shutil.py", line 121, in copyfile with open(dst, 'wb') as fdst: FileNotFoundError: [Errno 2] No such file or directory: '....\\database/8403e077-3fd5-4d65-9f45-ada5fab36f9b.jpg' — egj
– egj, Commented Mar 1, 2020 at 21:38
That error seems to be related to the file you are trying to move, not necessarily the folder you've created — Parakiwi
– Parakiwi, Commented Mar 1, 2020 at 22:17

alexfertel · Accepted Answer · 2020-03-01 23:43:09Z

4

I would recommend the following approach:

import os
from pathlib import Path

Path('path').mkdir(parents=True, exist_ok=True)

This works cross-platform and doesn't overwrite the directories if they already exist.

edited Mar 1, 2020 at 23:43

answered Mar 1, 2020 at 19:46

alexfertel

9461 gold badge10 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

erip Over a year ago

You don't need to os.path.join. You can just do Path("directory/directory").mkdir(...) as pathlib will normalize this.

erip Over a year ago

Yes it does, hence why I said "pathlib will normalize this." :-) Try it out: repl.it/repls/ExcitableTameSpools

alexfertel Over a year ago

Oh, wow, you're right! Let me delete the comment and edit the answer. Thanks!

Marcus · Accepted Answer · 2020-03-01 19:39:42Z

3

You should try the os module. It has a mkdir method that creates a directory based on the path you give it as a parameter.

import os
os.mkdir("path")

answered Mar 1, 2020 at 19:39

Marcus

3531 gold badge5 silver badges21 bronze badges

Comments

Marcin Orlowski · Accepted Answer · 2020-03-01 20:02:59Z

3

While most answers suggest using os.mkdir() I suggest you rather go for os.makedirs() which would recursively create all the missing folders in your path, which usually is more convinient.

import os
os.makedirs('foo/bar')

Docs: https://docs.python.org/3/library/os.html#os.makedirs

answered Mar 1, 2020 at 20:02

Marcin Orlowski

75.7k11 gold badges128 silver badges152 bronze badges

Collectives™ on Stack Overflow

How to create a folder in python?

3 Answers 3

3 Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

3 Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related