{"cells":[{"metadata":{"trusted":true},"cell_type":"code","source":"!cp /kaggle/input/decord/install.sh . && chmod  +x install.sh && ./install.sh ","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"import os, sys\nsys.path.insert(0,'/kaggle/working/reader/python')\n\nfrom decord import VideoReader\nfrom decord import cpu, gpu\nfrom decord.bridge import set_bridge\nimport glob","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"set_bridge('torch')","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"filenames = glob.glob('../input/deepfake-detection-challenge/test_videos/*.mp4')","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"### GPU version. Return torch.uint8 located on gpu. Have proble with memory"},{"metadata":{"trusted":true},"cell_type":"code","source":"%%time\n## Be carefull GPU memory leak\nshapes = []\nfor filename in filenames:\n    video = VideoReader(filename, ctx=gpu())\n    data = video.get_batch(range(len(video)))\n    shapes += [data.size()]\n    del video, data","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"### CPU version. Return torch.uint8"},{"metadata":{"trusted":true},"cell_type":"code","source":"%%time\n## slower but stable\nshapes = []\nfor filename in filenames:\n    video = VideoReader(filename, ctx=cpu())\n    data = video.get_batch(range(len(video)))\n    shapes += [data.size()]\n    del video, data","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"!rm -r reader && rm install.sh","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"!cp /kaggle/input/deepfake-detection-challenge/sample_submission.csv ./submission.csv","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat":4,"nbformat_minor":4}