Writing multiple images with OpenCv2 in Python

Question

I got a Data set which full of images. My aim is converting this images to gray scale. I can easily do this with one image but I need to do whit multiple images. Here is my code;

import cv2
import glob

path = "/path/*.jpg"

for file in glob.glob(path):
    img = cv2.imread(file)
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    cv2.imshow('Gray image', gray)
    cv2.imwrite('/path/cat' + '_gray.jpg', gray)

When I run this script, it's only convert a single image but my aim is converting the all images in the path. I'm really beginer on image processing and OpenCv2 module. Thanks already now for your helps.

This might help. Get a list of all your files first, then do your colour conversion (which you're doing correctly as far as I can tell). — Pam
– Pam, Commented Mar 19, 2020 at 16:31

Thomas · Accepted Answer · 2020-03-19 16:32:46Z

2

'/path/cat' + '_gray.jpg' always gives you /path/cat_gray.jpg, regardless of the input filename.

You probably want to use some part of the original file name there, which you can find using the functions in the os.path module.

For example:

import cv2
import glob
import os.path

path = "/path/*.jpg"

for file in glob.glob(path):
    img = cv2.imread(file)
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    cv2.imshow('Gray image', gray)
    basename = os.path.basename(file)  # e.g. MyPhoto.jpg
    name = os.path.splitext(basename)[0]  # e.g. MyPhoto
    cv2.imwrite('/path/' + name + '_gray.jpg', gray)

answered Mar 19, 2020 at 16:32

Thomas

183k57 gold badges383 silver badges510 bronze badges

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

Collectives™ on Stack Overflow

Writing multiple images with OpenCv2 in Python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related