- 24 Jan, 2022 1 commit
-
-
Prabhat Roy authored
-
- 19 Jan, 2022 1 commit
-
-
Prabhat Roy authored
* Return RGB frames as output of GPU decoder * Move clamp to the conversion function * Cleaned up a bit * Remove utility functions from test * Use data member width directly * Fix linter error
-
- 10 Jan, 2022 1 commit
-
-
Prabhat Roy authored
-
- 30 Dec, 2021 1 commit
-
-
Prabhat Roy authored
* [WIP] Add video GPU decoder * Expose use_dev_frame to python class and handle it internally * Fixed invalid argument CUDA error * Fixed empty and missing frames * Free remaining frames in the queue * Added nv12 to yuv420 conversion support for host frames * Added unit test and cleaned up code * Use CUDA_HOME inside if * Undo commented out code * Add Readme * Remove output_format and use_device_frame optional arguments from the VideoReader API * Cleaned up init() * Fix warnings * Fix python linter errors * Fix linter issues in setup.py * clang-format * Make reformat private * Member function naming * Add comments * Variable renaming * Code cleanup * Make return type of decode() void * Replace printing errors with throwing runtime_error * Replaced runtime_error with TORCH_CHECK in demuxer.h * Use CUDAGuard instead of cudaSetDevice * Remove printf * Use Tensor instead of uint8* and remove cuMemAlloc/cuMemFree * Use TORCH_CHECK instead of runtime_error * Use TORCHVISION_INCLUDE and TORCHVISION_LIBRARY to pass video codec location * Include ffmpeg_include_dir * Remove space * Removed use of runtime_error * Update Readme * Check for bsf.h * Change struct initialisation style * Clean-up get_operating_point * Make variable naming convention uniform * Move checking for bsf.h around * Fix linter error Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
- 21 Dec, 2021 1 commit
-
-
Kai Zhang authored
* add api usage log for io * cover VideoReader * cover c++ APIs * add api usage log for io * cover VideoReader * cover c++ APIs * add _cpp suffix to c++ APIs * use new API and change cpp format * remove _cpp suffix * adopt new API * lint
-
- 05 Dec, 2021 1 commit
-
-
Prabhat Roy authored
-
- 17 Nov, 2021 1 commit
-
-
F-G Fernandez authored
* style: Added typing annotations * style: Fixed lint * style: Fixed typing * chore: Updated mypy.ini * style: Fixed typing * chore: Updated mypy.ini * style: Fixed typing compatibility with jit * style: Fixed typing * style: Fixed typing * style: Fixed missing import * style: Fixed typing of __iter__ * style: Fixed typing * style: Fixed lint * style: Finished typing * style: ufmt the file * style: Removed unnecessary typing * style: Fixed typing of iterator Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
Prabhat Roy <prabhatroy@fb.com>
-
- 03 Nov, 2021 1 commit
-
-
Bruno Korbar authored
* modify processPacket to support fast seek * add fastSeek to ProcessPacket decoder definition * add fastseek flag to DecoderParametersStruct * add fastseek flag to the process packet call * no default params in C++ implementation * enable flag in C++ implementation * make order of parameters more normal * register new seek with python api * [somewhat broken] test suite for keyframes using pyav * revert " changes * add type annotations to init * Adding tests * linter * Flake doesn't show up :| * Change from unitest to pytest syntax * add return type Co-authored-by:
-
- 18 Oct, 2021 1 commit
-
-
Bruno Korbar authored
* initial commit * initial commit * flake8 * last nits Co-authored-by:
-
- 04 Oct, 2021 1 commit
-
-
Philip Meier authored
* add ufmt as code formatter * cleanup * quote ufmt requirement * split imports into more groups * regenerate circleci config * fix CI * clarify local testing utils section * use ufmt pre-commit hook * split relative imports into local category * Revert "split relative imports into local category" This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51. * pin black and usort dependencies * fix local test utils detection * fix ufmt rev * add reference utils to local category * fix usort config * remove custom categories sorting * Run pre-commit without fixing flake8 * got a double import in merge Co-authored-by:Nicolas Hug <nicolashug@fb.com>
-
- 31 Aug, 2021 1 commit
-
-
F-G Fernandez authored
* style: Added typing annotations * Specified types for iter and seek. Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Vasilis Vryniotis <vvryniotis@fb.com>
-
- 11 May, 2021 1 commit
-
-
Aditya Oke authored
* Fix io imports and expose methods * fix fmt Co-authored-by:
-
- 31 Mar, 2021 1 commit
-
-
Francisco Massa authored
This reverts commit 3bfdb427.
-
- 24 Mar, 2021 1 commit
-
-
Bruno Korbar authored
* multithreading allowed in stream codec context * numThreads is passed as a decoder parameter. At this stage code should be unchanged * enabling multithreading in videoReader API * moving defaults to header files * replace long with int64_t because torchscript * docstring for Num threads * Enable codec-related heuristics as defaults * Update torchvision/csrc/io/decoder/stream.cpp Co-authored-by:
-
- 26 Jan, 2021 1 commit
-
-
Nicolas Hug authored
Co-authored-by:
-
- 30 Oct, 2020 1 commit
-
-
Bruno Korbar authored
* Modified the example to conform with the new DICT return. * adding better stream documentation to examples * Clearing up some documentation as a result of a feedback * Formatting mostly * Addressing Victor's comments. * addressing fmassas comments * remove unnecessary tab Co-authored-by:
Bruno Korbar <korbar@vggdev9.vggdev.cluster> Co-authored-by:
vfdev <vfdev.5@gmail.com>
-
- 28 Oct, 2020 1 commit
-
-
So Uchida authored
-
- 13 Oct, 2020 1 commit
-
-
Bruno Korbar authored
* Video reader now returns dicts * docs update * Minor improvements Co-authored-by:
Bruno Korbar <bjuncek@Frazz.local> Co-authored-by:
-
- 12 Oct, 2020 1 commit
-
-
Francisco Massa authored
* Minor improvements to VideoReader * update jupyter notebook with new naming Co-authored-by:Bruno Korbar <bjuncek@gmail.com>
-
- 09 Oct, 2020 1 commit
-
-
Bruno Korbar authored
* initial API documentation attempt * test the docs * initial commit * updating test to match the registration * adding the warning on unsucessful import * Try to do conditional import * Simple fix? * clearing up docs * docstring commit * Adding types in arguments Co-authored-by:
-
- 06 Oct, 2020 1 commit
-
-
Edgar Andrés Margffoy Tuay authored
* Expose image io functions in docs * Expose image funcs in module * Update docstring * Update docstrings for all functions * Update section header
-
- 18 Sep, 2020 1 commit
-
-
Aditya Oke authored
* Adds docs and fixes improts for ops and io * reverts io and init for images
-
- 08 Jul, 2020 1 commit
-
-
Edgar Andrés Margffoy Tuay authored
* Add libpng requirement into conda recipe * Try to install libjpeg-turbo * Add PNG reading capabilities * Remove newline * Add image extension to compilation instructions * Include png functions as part of the main library * Update CMakeLists * Detect if building on conda-build * Debug * More debug messages * Print globbed libreries * Print globbed libreries * Point to correct PNG path * Remove libJPEG preventively * Debug extension loading * Link libpng explicitly * Link with PNG * Add PNG reading capabilities * Add libpng requirement into conda recipe * Try to install libjpeg-turbo * Remove newline * Add image extension to compilation instructions * Include png functions as part of the main library * Update CMakeLists * Detect if building on conda-build * Debug * More debug messages * Print globbed libreries * Print globbed libreries * Point to correct PNG path * Remove libJPEG preventively * Debug extension loading * Link libpng explicitly * Link with PNG * Install libpng on conda-based wheel distributions * Add -y flag * Add -y flag to yum * Locate LibPNG on windows conda * Remove empty else * Copy libpng16.so * Copy dylib on Mac * Improve check on Windows * Try to install ninja using conda on windows * Use libpng on Windows * Package lib on windows wheel * Point library to the correct place * Include binaries as part of wheel * Copy libpng.so on linux * Look for png.h on Windows when using conda-build * Do not skip png tests on Mac/Win * Restore libjpeg-turbo * Install jpeg-turbo on wheel distributions * Install libjpeg-turbo from conda-forge on wheel distributions * Do not pull av on conda-build * Add pillow disclaimer * Vendors libjpeg-turbo 2.0.4 * Merge JPEG work * Remove submodules * Regenerate circle config * Fix style issues * Fix C++ style issues * More style corrections * Add JPEG-turbo to linking libraries * More style corrections * More style corrections * More style corrections * Install libjpeg-turbo-devel * Install libturbo-jpeg on typing pipeline * Update Circle template * Windows and Unix turbojpeg have the same linking name * Install turbojpeg-devel instead of libjpeg-turbo * Copy TurboJPEG binaries to wheel * Move test image * Move back test image * Update JPEG test path * Remove dot from extension * Move image functions to extension * Use stdout arg in subprocess * Disable image extension if libpng or turbojpeg are not found * Append libpng stdout * Prevent list appending on lists * Minor path correction * Minor error correction * Add linking flags * Style issues correction * Address minor review corrections * Refactor library search * Restore access index * Fix JPEG tests * Update libpng version in Travis * Add -y flag * Remove dot * Update libpng using apt * Check libpng version * Change libturbojpeg binary * Update import * Change call * Restore av in conda recipe * Minor error correction * Remove unused comment in travis.yml * Update README * Fix missing links * Remove fixes for 16.04 * Enable JPEG support using libjpeg directly * Install libjpeg-turbo8 on Travis * Fix styling issues * Do not append to paths if library found in standard library locations * Add macro flag * Use custom error handler * Use 3? * Use short-lived buffer * Return TRUE instead of true * Assert RuntimeError * Use .jpg extension * Remove conda-forge * Use data_ptr instead of accessor * Use assertTrue for jpeg verification * Remove unnecessary memcpy * Debug test on Windows * Remove PIL from jpeg verification Co-authored-by:Ryad ZENINE <r.zenine@gmail.com>
-
- 02 Jul, 2020 2 commits
-
-
Francisco Massa authored
This reverts commit c1a99b7b.
-
Edgar Andrés Margffoy Tuay authored
* Add libpng requirement into conda recipe * Try to install libjpeg-turbo * Add PNG reading capabilities * Remove newline * Add image extension to compilation instructions * Include png functions as part of the main library * Update CMakeLists * Detect if building on conda-build * Debug * More debug messages * Print globbed libreries * Print globbed libreries * Point to correct PNG path * Remove libJPEG preventively * Debug extension loading * Link libpng explicitly * Link with PNG * Add PNG reading capabilities * Add libpng requirement into conda recipe * Try to install libjpeg-turbo * Remove newline * Add image extension to compilation instructions * Include png functions as part of the main library * Update CMakeLists * Detect if building on conda-build * Debug * More debug messages * Print globbed libreries * Print globbed libreries * Point to correct PNG path * Remove libJPEG preventively * Debug extension loading * Link libpng explicitly * Link with PNG * Install libpng on conda-based wheel distributions * Add -y flag * Add -y flag to yum * Locate LibPNG on windows conda * Remove empty else * Copy libpng16.so * Copy dylib on Mac * Improve check on Windows * Try to install ninja using conda on windows * Use libpng on Windows * Package lib on windows wheel * Point library to the correct place * Include binaries as part of wheel * Copy libpng.so on linux * Look for png.h on Windows when using conda-build * Do not skip png tests on Mac/Win * Restore libjpeg-turbo * Install jpeg-turbo on wheel distributions * Install libjpeg-turbo from conda-forge on wheel distributions * Do not pull av on conda-build * Add pillow disclaimer * Vendors libjpeg-turbo 2.0.4 * Merge JPEG work * Remove submodules * Regenerate circle config * Fix style issues * Fix C++ style issues * More style corrections * Add JPEG-turbo to linking libraries * More style corrections * More style corrections * More style corrections * Install libjpeg-turbo-devel * Install libturbo-jpeg on typing pipeline * Update Circle template * Windows and Unix turbojpeg have the same linking name * Install turbojpeg-devel instead of libjpeg-turbo * Copy TurboJPEG binaries to wheel * Move test image * Move back test image * Update JPEG test path * Remove dot from extension * Move image functions to extension * Use stdout arg in subprocess * Disable image extension if libpng or turbojpeg are not found * Append libpng stdout * Prevent list appending on lists * Minor path correction * Minor error correction * Add linking flags * Style issues correction * Address minor review corrections * Refactor library search * Restore access index * Fix JPEG tests * Update libpng version in Travis * Add -y flag * Remove dot * Update libpng using apt * Check libpng version * Change libturbojpeg binary * Update import * Change call * Restore av in conda recipe * Minor error correction * Remove unused comment in travis.yml * Update README * Fix missing links * Remove fixes for 16.04 * Remove JPEG-related code * Remove installation references to turbojpeg * Remove further references to turbojpeg * Fix c++ style issues * Fix c++ style issues * Fix libpng-config include flag parsing * Remove conda-forge * Remove include dirs from main extension * Do not pass extra include and library paths to main torchvision extension * Add libpng to environment.yml * Remove inexistent imports * Add instructions regarding environment variables to README Co-authored-by:
-
- 01 Jul, 2020 1 commit
-
-
Edgar Andrés Margffoy Tuay authored
This reverts commit 766721b1.
-
- 30 Jun, 2020 1 commit
-
-
Edgar Andrés Margffoy Tuay authored
* Add libpng requirement into conda recipe * Try to install libjpeg-turbo * Add PNG reading capabilities * Remove newline * Add image extension to compilation instructions * Include png functions as part of the main library * Update CMakeLists * Detect if building on conda-build * Debug * More debug messages * Print globbed libreries * Print globbed libreries * Point to correct PNG path * Remove libJPEG preventively * Debug extension loading * Link libpng explicitly * Link with PNG * Add PNG reading capabilities * Add libpng requirement into conda recipe * Try to install libjpeg-turbo * Remove newline * Add image extension to compilation instructions * Include png functions as part of the main library * Update CMakeLists * Detect if building on conda-build * Debug * More debug messages * Print globbed libreries * Print globbed libreries * Point to correct PNG path * Remove libJPEG preventively * Debug extension loading * Link libpng explicitly * Link with PNG * Install libpng on conda-based wheel distributions * Add -y flag * Add -y flag to yum * Locate LibPNG on windows conda * Remove empty else * Copy libpng16.so * Copy dylib on Mac * Improve check on Windows * Try to install ninja using conda on windows * Use libpng on Windows * Package lib on windows wheel * Point library to the correct place * Include binaries as part of wheel * Copy libpng.so on linux * Look for png.h on Windows when using conda-build * Do not skip png tests on Mac/Win * Restore libjpeg-turbo * Install jpeg-turbo on wheel distributions * Install libjpeg-turbo from conda-forge on wheel distributions * Do not pull av on conda-build * Add pillow disclaimer * Vendors libjpeg-turbo 2.0.4 * Merge JPEG work * Remove submodules * Regenerate circle config * Fix style issues * Fix C++ style issues * More style corrections * Add JPEG-turbo to linking libraries * More style corrections * More style corrections * More style corrections * Install libjpeg-turbo-devel * Install libturbo-jpeg on typing pipeline * Update Circle template * Windows and Unix turbojpeg have the same linking name * Install turbojpeg-devel instead of libjpeg-turbo * Copy TurboJPEG binaries to wheel * Move test image * Move back test image * Update JPEG test path * Remove dot from extension * Move image functions to extension * Use stdout arg in subprocess * Disable image extension if libpng or turbojpeg are not found * Append libpng stdout * Prevent list appending on lists * Minor path correction * Minor error correction * Add linking flags * Style issues correction * Address minor review corrections * Refactor library search * Restore access index * Fix JPEG tests * Update libpng version in Travis * Add -y flag * Remove dot * Update libpng using apt * Check libpng version * Change libturbojpeg binary * Update import * Change call * Restore av in conda recipe * Minor error correction * Remove unused comment in travis.yml * Update README * Fix missing links * Remove fixes for 16.04 Co-authored-by:
-
- 09 Apr, 2020 1 commit
-
-
Francisco Massa authored
Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/2077 Pull Request resolved: https://github.com/facebookresearch/SlowFast/pull/164 This is a follow-up diff from D18720474 We will be releasing a new version of torchvision soon and the signature of those functions is not ready yet, following my comment in https://our.intern.facebook.com/intern/diff/D18720474/?transaction_id=561239541337402 Reviewed By: stephenyan1231 Differential Revision: D20914571 fbshipit-source-id: 1a7560b8f8e46ab42ef376c50b494a4f73923e94 Co-authored-by:
Francisco Massa <fmassa@fb.com>
-
- 28 Jan, 2020 1 commit
-
-
Francisco Massa authored
* torchscriptable functions for video io (#1653) Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/1653 created new torchscriptable video io functions as part of the api: read_video_meta_data_from_memory and read_video_from_memory. Updated the implementation of some of the internal functions to be torchscriptable. Reviewed By: stephenyan1231 Differential Revision: D18720474 fbshipit-source-id: 4ee646b66afecd2dc338a71fd8f249f25a3263bc * BugFix Co-authored-by:
Jon Guerin <54725679+jguerin-fb@users.noreply.github.com>
-
- 23 Oct, 2019 1 commit
-
-
Francisco Massa authored
* Unify video backend interfaces * Remove reference cycle * Make functions private and enable tests on OSX * Disable test if video_reader backend not available * Lint * Fix import after refactoring * Fix lint
-
- 12 Oct, 2019 1 commit
-
-
Zhicheng Yan authored
* extend video reader to support fast video probing * fix c++ lint * small fix * allow to accept input video of type torch.Tensor
-
- 08 Oct, 2019 1 commit
-
-
Zhicheng Yan authored
-
- 03 Oct, 2019 1 commit
-
-
Zhicheng Yan authored
* add metadata to video dataset classes. bug fix. more robustness * query video backend within VideoClips class * Fix tests * Fix lint
-
- 24 Sep, 2019 1 commit
-
-
Zhicheng Yan authored
* add _backend argument to __init__() of class VideoClips * minor fix * minor fix * Make backend private in VideoClips * Fix lint
-
- 20 Sep, 2019 1 commit
-
-
Zhicheng Yan authored
* [video reader] inception commit * add method save_metadata to class VideoClips in video_utils.py * add load_metadata() method to VideoClips class * add Exception to not catch unexpected events such as memory erros, interrupt * fix bugs in video_plus.py * [video reader]remove logging. update setup.py * remove time measurement in test_video_reader.py * Remove glog and try making ffmpeg finding more robust * Add ffmpeg to conda build * Add ffmpeg to conda build [again] * Make library path finding more robust * Missing import * One more missing fix for import * Py2 compatibility and change package to av to avoid version conflict with ffmpeg * Fix for python2 * [video reader] support to decode one stream only (e.g. video/audio stream) * remove argument _precomputed_metadata_filepath * remove save_metadata method * add get_metadata method * expose _precomputed_metadata and frame_rate arguments in video dataset __init__ method * remove ssize_t * remove size_t to pass CI check on Windows * add PyInit__video_reader function to pass CI check on Windows * minor fix to define PyInit_video_reader symbol * Make c++ video reader optional * Temporarily revert changes to test_io * Revert changes to python files * Rename files to make it private * Fix python lint * Fix C++ lint * add a functor object EnumClassHash to make Enum class instances usable as key type of std::unordered_map * fix cpp format check
-
- 02 Jul, 2019 1 commit
-
-
Francisco Massa authored
* WIP * WIP * Add some documentation * Improve tests and add GC collection * [WIP] add timestamp getter * Bugfixes * Improvements and travis * Add audio fine-grained alignment * More doc * Remove unecessary file * Remove comment * Lazy import av * Remove hard-coded constants for the test * Return info stats from read * Fix for Python-2
-
