Add video classification pipeline (#20151)

* 🚧 wip video classification pipeline

* 🚧 wip - add is_decord_available check

* 🐛 add missing import

*  add tests

* 🔧 add decord to setup extras

* 🚧 add is_decord_available

*  add video-classification pipeline

* 📝 add video classification pipe to docs

* 🐛 add missing VideoClassificationPipeline import

* 📌 add decord install in test runner

*  fix url inputs to video-classification pipeline

*  updates from review

* 📝 add video cls pipeline to docs

* 📝 add docstring

* 🔥 remove unused import

* 🔥 remove some code

* 📝 docfix
This commit is contained in:
Nathan Raw
2022-12-08 16:22:43 -05:00
committed by GitHub
parent c56ebbbea6
commit 9e56aff58a
12 changed files with 272 additions and 3 deletions

View File

@@ -103,6 +103,7 @@ _deps = [
"cookiecutter==1.7.3",
"dataclasses",
"datasets!=2.5.0",
"decord==0.6.0",
"deepspeed>=0.6.5",
"dill<0.3.5",
"evaluate>=0.2.0",
@@ -286,7 +287,7 @@ extras["vision"] = deps_list("Pillow")
extras["timm"] = deps_list("timm")
extras["natten"] = deps_list("natten")
extras["codecarbon"] = deps_list("codecarbon")
extras["video"] = deps_list("decord")
extras["sentencepiece"] = deps_list("sentencepiece", "protobuf")
extras["testing"] = (
@@ -332,6 +333,7 @@ extras["all"] = (
+ extras["timm"]
+ extras["codecarbon"]
+ extras["accelerate"]
+ extras["video"]
)
# Might need to add doc-builder and some specific deps in the future