Skip to content

Commit

Permalink
Adição do exemplo de classificação de vídeo utilizando opencv
Browse files Browse the repository at this point in the history
  • Loading branch information
M3nin0 committed Jan 24, 2019
0 parents commit fb7c0aa
Show file tree
Hide file tree
Showing 3 changed files with 249 additions and 0 deletions.
204 changes: 204 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,204 @@

# Created by https://www.gitignore.io/api/node,python
# Edit at https://www.gitignore.io/?templates=node,python

### Node ###
# Logs
logs
*.log
npm-debug.log*
yarn-debug.log*
yarn-error.log*

# Runtime data
pids
*.pid
*.seed
*.pid.lock

# Directory for instrumented libs generated by jscoverage/JSCover
lib-cov

# Coverage directory used by tools like istanbul
coverage

# nyc test coverage
.nyc_output

# Grunt intermediate storage (https://gruntjs.com/creating-plugins#storing-task-files)
.grunt

# Bower dependency directory (https://bower.io/)
bower_components

# node-waf configuration
.lock-wscript

# Compiled binary addons (https://nodejs.org/api/addons.html)
build/Release

# Dependency directories
node_modules/
jspm_packages/

# TypeScript v1 declaration files
typings/

# Optional npm cache directory
.npm

# Optional eslint cache
.eslintcache

# Optional REPL history
.node_repl_history

# Output of 'npm pack'
*.tgz

# Yarn Integrity file
.yarn-integrity

# dotenv environment variables file
.env
.env.test

# parcel-bundler cache (https://parceljs.org/)
.cache

# next.js build output
.next

# nuxt.js build output
.nuxt

# vuepress build output
.vuepress/dist

# Serverless directories
.serverless/

# FuseBox cache
.fusebox/

# DynamoDB Local files
.dynamodb/

### Python ###
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class

# C extensions
*.so

# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
share/python-wheels/
*.egg-info/
.installed.cfg
*.egg
MANIFEST

# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec

# Installer logs
pip-log.txt
pip-delete-this-directory.txt

# Unit test / coverage reports
htmlcov/
.tox/
.nox/
.coverage
.coverage.*
nosetests.xml
coverage.xml
*.cover
.hypothesis/
.pytest_cache/

# Translations
*.mo
*.pot

# Django stuff:
local_settings.py
db.sqlite3

# Flask stuff:
instance/
.webassets-cache

# Scrapy stuff:
.scrapy

# Sphinx documentation
docs/_build/

# PyBuilder
target/

# Jupyter Notebook
.ipynb_checkpoints

# IPython
profile_default/
ipython_config.py

# pyenv
.python-version

# celery beat schedule file
celerybeat-schedule

# SageMath parsed files
*.sage.py

# Environments
.venv
env/
venv/
ENV/
env.bak/
venv.bak/

# Spyder project settings
.spyderproject
.spyproject

# Rope project settings
.ropeproject

# mkdocs documentation
/site

# mypy
.mypy_cache/
.dmypy.json
dmypy.json

# Pyre type checker
.pyre/

### Python Patch ###
.venv/

# End of https://www.gitignore.io/api/node,python
45 changes: 45 additions & 0 deletions video-process-opencv/app.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
'''
Script para a classificação de vídeos utilizando os pesos do retreino
feito para identificar gestos de Libras
'''

import cv2 as cv

import numpy as np

from keras.models import load_model
from keras.applications.mobilenet import preprocess_input

model = load_model('weights/results_ican.h5')

CLASSES = ['Amigo', 'Desculpa', 'Telefone']


def classify(img_file):
r = model.predict(prepare_image(img_file))
return CLASSES[np.argmax(r)]


def prepare_image(arr):
arr = cv.resize(arr, (224, 224))
img_array_expanded_dims = np.expand_dims(arr, axis=0)
return preprocess_input(img_array_expanded_dims)

video = cv.VideoCapture(0)

while True:
ret, frame = video.read()

c = classify(frame)

frame = cv.putText(frame, c, (10, 40), cv.FONT_HERSHEY_COMPLEX, 1, 128)
cv.imshow('Video de classificação', frame)

if cv.waitKey(1) & 0xFF == ord('q'):
break

video.release()
cv.destroyAllWindows()

Binary file added video-process-opencv/weights/results_ican.h5
Binary file not shown.

0 comments on commit fb7c0aa

Please sign in to comment.