A story of using rembg to make a white background + person video into a black background video

Overview

Using ffmpeg, rembg, opencv, I changed the video of white background + person to black background + person video.

background

For the wedding entertainment, I decided to make a parody video of perfume. I wanted to make a video of people singing on a black background like a music video for perfume, but I didn't want to spend time buying a black cloth. She wanted to make the white background somehow black because she could shoot anywhere on a white background.

What it was like

raw.jpg This is,

3.jpg It looks like this. Great accuracy. rembg is amazing. I tried various other person cropping methods, but this was the most natural.

Premise

--python is installed. --rembg is installed. (This installation is troublesome. Maybe I can explain it separately) --opencv is installed. --ffmpeg is installed and PATH is in place.

procedure

--Preparation of video files (MP4, etc., do not raise FPS more than necessary (processing becomes heavy)) --Disassemble video files into image files.

Command example (video frame image is output to raw folder): ffmpeg -i ~~ .mp4 -vcodec png raw \ image_% 05d.png

--Crop the person in the image file using rembg. It takes a lot of time because it targets all the images in the video.

Command example: rembg -a -ae 15 -o output1 \ image_252.png raw \ image_00001.png

https://github.com/danielgatis/rembg However, it was quite difficult to introduce. Difficult version of pytorch, torchvision, etc. I think it is better to build it in the virtual environment of pyenv or conda.

--Fill the transparent part after cropping the person with a different image or any color. I did it with opencv.

--Animate the image

Command example: ffmpeg -f image2 -r 30 -i image_% 03d.png -r 30 -an -vcodec libx264 -pix_fmt yuv420p video.mp4

Script used

Folder structure

The folder structure of the script used this time is as follows.

root/
 ├ raw_video/ #Folder to save raw videos
 ├ formatted_video/ #toMP4.Folder to save FPS and video quality adjusted with py
 ├ output/ #Folder to save the final generated video
 ├ mask/ #Folder to save the black background image
    └ black.png #A 1920x1080 black image is prepared according to the size of the video. It doesn't matter if it's not black.
 ├ tmp1/ # formatted_Folder to save the image of the video
 ├ tmp2/ #Folder to save the image of tmp1 with the white background transparent with rembg
 ├ tmp3/ #tmp2 image and mask/black.Folder to save png images with opencv and black background
 ├ toMP4.py # raw_Adjust the FPS and image quality of the video video, formatted_Script to put in the video folder
 └ main.py #Script that performs a series of processing

Code to make a full HD, 30FPS MP4 file (toMP4.py)

A script that generates ffmpeg commands in python and executes them.

import os

base_dir = "raw_video"
output_dir = "formatted_video"
fns = os.listdir(base_dir)

print(len(fns))
for f in fns:
    print(f)
    cmd = "ffmpeg.exe -i {} -s hd1080 -c:v libx264 -c:a copy -r 30 {} -y".format(
        base_dir + "\\" + f, output_dir + "\\" + f + "__.mp4")
    os.system(cmd)

Code that performs a series of processing (main.py)

--Disassemble video files into image files. --Crop the person in the image file using rembg. It takes a lot of time because it targets all the images in the video. --Fill the transparent part after cropping the person with a different image or any color. --Animate the image

import os
import shutil
import cv2
import matplotlib.pylab as plt

base_dir = "formatted_video"
tmp1_dir = "tmp1"
tmp2_dir = "tmp2"
tmp3_dir = "tmp3"
output_dir = "output"
fns = os.listdir(base_dir)

print(len(fns))
for f in fns:
    bn = os.path.basename(f)
    print(f)
    #Create folder
    shutil.rmtree(tmp1_dir)
    os.makedirs(tmp1_dir)
    shutil.rmtree(tmp2_dir)
    os.makedirs(tmp2_dir)
    shutil.rmtree(tmp3_dir)
    os.makedirs(tmp3_dir)
    # os.makedirs(output_dir)

    # #Break down video files into image files
    p1 = base_dir + "\\" + f
    p2 = tmp1_dir + "\\" + bn + "_%05d.png "
    cmd1 = "ffmpeg -i {} -vcodec png {}".format(p1, p2)
    print(cmd1)
    os.system(cmd1)
    #Crop a person in an image file using rembg
    fn2s = os.listdir(tmp1_dir)
    for f2 in fn2s:
        if(f2[-3:] != "png"):
            continue
        p3 = tmp1_dir + "\\" + f2
        p4 = tmp2_dir + "\\" + f2
        cmd2 = "rembg -a -ae 15 -o {} {}".format(p4, p3)
        print(cmd2)
        os.system(cmd2)
    #Fill the transparent part after cropping the person with a different image or any color
    fn2s = os.listdir(tmp2_dir)
    for f2 in fn2s:
        if(f2[-3:] != "png"):
            continue
        print('{}/{}'.format(tmp2_dir, f2))
        frame = cv2.imread('mask/black1.png')
        png_image = cv2.imread('{}/{}'.format(tmp2_dir, f2),
                               cv2.IMREAD_UNCHANGED)  #Read with Alfa channel included
        # png_image[:, :, 3:] = np.where(png_image[:, :, 3:] > 200, 255, 0)
        x1, y1, x2, y2 = 0, 0, png_image.shape[1], png_image.shape[0]
        frame[y1:y2, x1:x2] = frame[y1:y2, x1:x2] * (1 - png_image[:, :, 3:] / 255) + \
            png_image[:, :, :3] * (png_image[:, :, 3:] / 255)
        # plt.imshow(frame)
        # plt.show()
        cv2.imwrite('{}/{}'.format(tmp3_dir, f2), frame)
    #Animate the image
    cmd3 = "ffmpeg -f image2 -r 30 -i {} -r 30 -an -vcodec libx264 -pix_fmt yuv420p {}.mp4 -y".format(
        tmp3_dir + "\\" + bn + "_%05d.png ", output_dir + "\\" + bn + "_bg_blk")
    os.system(cmd3)
    # break

It also contains extra code for debugging.

install rembg

--Create a virtual environment with Anaconda. (Python = 3.8)

Finally

Please note that the content is only about your own memo, so it may be difficult to understand.

Recommended Posts

A story of using rembg to make a white background + person video into a black background video
I tried to make a regular expression of "amount" using Python
I tried to make a regular expression of "time" using Python
I tried to make a regular expression of "date" using Python
How to save only a part of a long video using OpenCV
Try to make a kernel of Jupyter
[Python] Smasher tried to make the video loading process a function using a generator
I tried to make a suspicious person MAP quickly using Geolonia address data
I tried to make a ○ ✕ game using TensorFlow
The story of introducing a multi-factor authentication function using a one-time password into a Java application
How to make a Python package using VS Code
Basics of PyTorch (2) -How to make a neural network-
The story of using circleci to build manylinux wheels
A story that contributes to new corona analysis using a free trial of Google Cloud Platform
I tried using PI Fu to generate a 3D model of a person from one image
[Python] How to make a list of character strings character by character
Make a given number of seconds into hours, minutes and seconds
Convert video to black and white with ffmpeg + python + opencv
The story of creating a database using the Google Analytics API
I tried to make a simple text editor using PyQt
[End of 2020] A memo to start using AWS CLI (Version 2)
How to connect the contents of a list into a string
[Python] A story that seemed to fall into a rounding trap
[Twitter] I want to make the downloaded past tweets (of my account) into a beautiful CSV
A story about a person who uses Python addicted to the judgment of an empty JavaScript dictionary