CODEWITHSUNDEEP
pythonimageopencvvideoresize
GGzet
GGzet

Reputation: 329

How to resize frame's from video with aspect ratio

I am using Python 2.7, OpenCV. I have written this code. 

import cv2
vidcap = cv2.VideoCapture('myvid2.mp4')
success,image = vidcap.read()
count = 0;
print "I am in success"
while success:
  success,image = vidcap.read()
  resize = cv2.resize(image, (640, 480)) 
  cv2.imwrite("%03d.jpg" % count, resize)     
  if cv2.waitKey(10) == 27:                     
      break
  count += 1

I am working with video and am dividing the video into individual frames, as a .jpg images. I am also at the same time resizing the frames to dimension 640x480. The order of the frames is also being preserved. The only issue with the code is that it does not save the previous image-ratio.

For example how it look's like, resize from 1920x1080: Resized image by pure code

There is a problem in ratio, as you can see. 1920x1080 16:9, but 640:480 4:3

How I ideally want it to be: Resized image in ideal code

Thank you for your taking the time for reading the question. I will be very glad if you can help me solve this issue~ Have a good day, my friend.

Upvotes: 8

Views: 55147

Answers (3)

Saransh Kejriwal
Saransh Kejriwal

Reputation: 2533

Instead of using hard-coded values 640 and 480, you can divide the original frame height and width by a value and supply that as an argument, like so:

import cv2

vidcap = cv2.VideoCapture("/path/to/video")
success, image = vidcap.read()
count = 0

while success:
    height, width, layers = image.shape
    new_h = height / 2
    new_w = width / 2
    resize = cv2.resize(image, (new_w, new_h))
    cv2.imwrite("%03d.jpg" % count, resize)
    success, image = vidcap.read()
    count += 1

Upvotes: 14

1Ghasthunter1
1Ghasthunter1

Reputation: 131

Wanted to add onto what Saransh mentioned about dividing. Dividing works great, but I believe the resize function expects that the new dimensions be an int object, so make sure you use the int(x) or round(x, 0) function to do so.

Upvotes: 2

Jobin Mathew
Jobin Mathew

Reputation: 29

Please try this.. It should give you the expected output.

 def resize_image(image, width, height,COLOUR=[0,0,0]):
    h, w, layers = image.shape
    if h > height:
        ratio = height/h
        image = cv2.resize(image,(int(image.shape[1]*ratio),int(image.shape[0]*ratio)))
    h, w, layers = image.shape
    if w > width:
        ratio = width/w
        image = cv2.resize(image,(int(image.shape[1]*ratio),int(image.shape[0]*ratio)))
    h, w, layers = image.shape
    if h < height and w < width:
        hless = height/h
        wless = width/w
        if(hless < wless):
            image = cv2.resize(image, (int(image.shape[1] * hless), int(image.shape[0] * hless)))
        else:
            image = cv2.resize(image, (int(image.shape[1] * wless), int(image.shape[0] * wless)))
    h, w, layers = image.shape
    if h < height:
        df = height - h
        df /= 2
        image = cv2.copyMakeBorder(image, int(df), int(df), 0, 0, cv2.BORDER_CONSTANT, value=COLOUR)
    if w < width:
        df = width - w
        df /= 2
        image = cv2.copyMakeBorder(image, 0, 0, int(df), int(df), cv2.BORDER_CONSTANT, value=COLOUR)
    image = cv2.resize(image,(1280,720),interpolation=cv2.INTER_AREA)
    return image

Upvotes: 1

Related Questions