added base
parent
d647d0981b
commit
d3b8971102
|
@ -0,0 +1,77 @@
|
|||
# Created by .ignore support plugin (hsz.mobi)
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
*.so
|
||||
.Python
|
||||
build/
|
||||
develop-eggs/
|
||||
dist/
|
||||
downloads/
|
||||
eggs/
|
||||
.eggs/
|
||||
lib/
|
||||
lib64/
|
||||
parts/
|
||||
sdist/
|
||||
var/
|
||||
wheels/
|
||||
share/python-wheels/
|
||||
*.egg-info/
|
||||
.installed.cfg
|
||||
*.egg
|
||||
MANIFEST
|
||||
*.manifest
|
||||
*.spec
|
||||
pip-log.txt
|
||||
pip-delete-this-directory.txt
|
||||
htmlcov/
|
||||
.tox/
|
||||
.nox/
|
||||
.coverage
|
||||
.coverage.*
|
||||
.cache
|
||||
nosetests.xml
|
||||
coverage.xml
|
||||
*.cover
|
||||
*.py,cover
|
||||
.hypothesis/
|
||||
.pytest_cache/
|
||||
cover/
|
||||
*.mo
|
||||
*.pot
|
||||
*.log
|
||||
local_settings.py
|
||||
db.sqlite3
|
||||
db.sqlite3-journal
|
||||
instance/
|
||||
.webassets-cache
|
||||
.scrapy
|
||||
docs/_build/
|
||||
.pybuilder/
|
||||
target/
|
||||
.ipynb_checkpoints
|
||||
profile_default/
|
||||
ipython_config.py
|
||||
__pypackages__/
|
||||
celerybeat-schedule
|
||||
celerybeat.pid
|
||||
*.sage.py
|
||||
.env
|
||||
.venv
|
||||
env/
|
||||
venv/
|
||||
ENV/
|
||||
env.bak/
|
||||
venv.bak/
|
||||
.spyderproject
|
||||
.spyproject
|
||||
.ropeproject
|
||||
/site
|
||||
.mypy_cache/
|
||||
.dmypy.json
|
||||
dmypy.json
|
||||
.pyre/
|
||||
.pytype/
|
||||
cython_debug/
|
||||
/data/
|
|
@ -0,0 +1,59 @@
|
|||
import cv2
|
||||
import os
|
||||
import math
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plt
|
||||
|
||||
# Create the temp dir
|
||||
if not os.path.exists('data'):
|
||||
os.makedirs('data')
|
||||
|
||||
# Define the color spaces
|
||||
Red_lower2 = np.array([,100,100])
|
||||
Red_upper2 = np.array([,255,255])
|
||||
Red_lower1 = np.array([0,100,100])
|
||||
Red_upper1 = np.array([6,255,255])
|
||||
Orange_lower = np.array([7,100,100])
|
||||
Orange_upper = np.array([22,255,255])
|
||||
Yellow_lower = np.array([23,100,100])
|
||||
Yellow_upper = np.array([32,255,255])
|
||||
Green_lower = np.array([33,100,100])
|
||||
Green_upper = np.array([75,255,255])
|
||||
Cyan_lower = np.array([76,100,100])
|
||||
Cyan_upper = np.array([97,255,255])
|
||||
Blue_lower = np.array([93,100,100])
|
||||
Blue_upper = np.array([130,255,255])
|
||||
Purple_lower = np.array([130,100,100])
|
||||
Purple_upper = np.array([,255,255])
|
||||
Pink_lower = np.array([,100,100])
|
||||
Pink_upper = np.array([,255,255])
|
||||
White_lower = np.array([,100,100])
|
||||
White_upper = np.array([,255,255])
|
||||
Black_lower = np.array([,100,100])
|
||||
Black_upper = np.array([,255,255])
|
||||
|
||||
# loop through video files
|
||||
videopath = '/media/vm/HDD/freeloops/3D/'
|
||||
imagepath = 'data/'
|
||||
videos = os.listdir(videopath)
|
||||
for video in videos:
|
||||
cap = cv2.VideoCapture(videopath + video)
|
||||
framerate = cap.get(cv2.CAP_PROP_FPS)
|
||||
while cap.isOpened():
|
||||
frameId = cap.get(1)
|
||||
ret, frame = cap.read()
|
||||
if not ret:
|
||||
break
|
||||
if frameId % math.floor(framerate) == 0:
|
||||
filename = imagepath + "/screenshot_" + str(int(frameId)) + ".jpg"
|
||||
cv2.imwrite(filename, frame)
|
||||
del cap
|
||||
# analyze each screenshot for colors
|
||||
images = os.listdir(imagepath)
|
||||
for image in images:
|
||||
cap = cv2.imread(imagepath + image, cv2.COLOR_BGR2HSV)
|
||||
plt.imshow(cap)
|
||||
|
||||
|
||||
|
||||
# move video file according to results
|
Loading…
Reference in New Issue