# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: Python 3
#     language: python
#     name: python3
# ---

import sys
from yolo import YOLO, detect_video
from PIL import Image

# +
# the input file path
input_path = "/data/dataset/relaxing_highway_traffic_5992.mp4"
output_path = "output_traffic.mp4"

FLAGS = dict()
FLAGS = {"model_path": "model_data/yolo_tiny.h5",
         "anchors_path": "model_data/tiny_yolo_anchors.txt",
         "classes_path": "model_data/coco_classes.txt"}

detect_video(YOLO(FLAGS), input_path, output_path)
# -

print(python.__dir__)

StarcoderdataJupyterScriptsDedupFiltered