[PYTHON] Extraits de code souvent utilisés lors du traitement de vidéos avec Google Colaboratory

Monter Google Drive

gdrive_path = "/content/gdrive" #Chemin monté

from google.colab import drive
drive.mount(gdrive_path)

Itération

Bouclez n'importe quelle plage à n'importe quel intervalle

for a in range(0, 1280, 10):
  print(a)

Répertorier les fichiers image dans un répertoire spécifique

import glob
files = glob.glob("/content/gdrive/My Drive/ColabExp/*.png ") #Seuls les fichiers avec l'extension png
files.sort() #Trier les fichiers trouvés

Créez un répertoire temporaire et travaillez

import tempfile
with tempfile.TemporaryDirectory() as tmpdir:
  #Ecrivez le fichier dans tmpdir

Ou

import tempfile
tmpdir = tempfile.TemporaryDirectory().name

Obtenez le nom du fichier à partir du chemin du fichier

import os
os.path.basename(files[i])

Afficher la progression du traitement

from tqdm import tqdm
for i, file in enumerate(tqdm(src_files)):
  #En traitement

Traitement d'image

Faire une toile avec PIL

from PIL import Image
img = Image.new('RGB', (1280, 720), (255, 255, 255)) # 1280 x 720px

Dessinez une figure avec PIL

from PIL import ImageDraw

draw = ImageDraw.Draw(img)

draw.rectangle(((x0, y0), (x1, y1)), fill=(255, 0, 0)) #Carré

Enregistrer des images avec OpenCV

import cv2
cv2.imwrite(save_path, image)

Enregistrer des images avec PIL / Pillow

img.save(path)

Transformez un tableau d'images en vidéo

!ffmpeg -framerate 30 -i "$dst_img_dir/%06d.png " -vcodec h264 -pix_fmt yuv420p -vf "scale=trunc(iw/2)*2:trunc(ih/2)*2" "$dst_mov_dir/out.mp4"

CSV

Lire CSV à partir d'un fichier (DataFrame)

df = pd.read_csv(filepath)

Incorporer et lire CSV (DataFrame)

CSV_DATA = StringIO("""x,y
10.0, 20.0
15.0, 30.0
20.0, 45.0
35.0, 80.0
""")

df = pd.read_csv(CSV_DATA, sep=",")

Enregistrer le contenu du DataFrame dans un fichier CSV

df.to_csv("out.csv")

Heures du jour

Horodatage de l'heure actuelle en secondes

from datetime import datetime
now_ts = int(datetime.now().timestamp())