• 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
setup.py 20.1 KB