1. 24 Jan, 2022 1 commit
  2. 19 Jan, 2022 1 commit
  3. 30 Dec, 2021 1 commit
    • Prabhat Roy's avatar
      Add video GPU decoder (#5019) · 64d21d12
      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: default avatarFrancisco Massa <fvsmassa@gmail.com>
      64d21d12