Как прочитать видео файл из местоположения sftp, используя opencv в python - PullRequest
Новая
       16

Как прочитать видео файл из местоположения sftp, используя opencv в python

0 голосов
/ 22 апреля 2019

Я сталкиваюсь с проблемой при чтении файла из местоположения sftp с использованием opencv lib. Можете ли вы сказать мне, как читать файл из местоположения sftp или объекта файла sftp. Если вы можете сказать мне читать большие файлы напрямую в opencv lib, то это хорошо. 

import paramiko
import cv2
import numpy as np

client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect("IPADDRESS", port=22, username='USERNAME', password='PASSWORD')
t = client.get_transport()
sftp = paramiko.SFTPClient.from_transport(t)
sftp.chdir("/home/bizviz/devanshu_copy")

obj = sftp.open("SampleVideo_1280x720_1mb.mp4")

cap = cv2.VideoCapture.open(obj)

while True:
    _,frame = cap.read()
    print(frame)
    cv2.imshow('res', frame)
    key = cv2.waitKey(1)
    if key == 27:
        break

cap.release()
cv2.destroyAllWindows()

1 Ответ

2 голосов
/ 22 апреля 2019

используя только Paramiko, вам нужно скопировать файл в локальную файловую систему, а затем использовать этот локальный файл для cv2.

cv2 не принимает этот метод передачи файла.

Конечно, в python есть библиотеки для всего, поэтому я думаю, что использование fs.sshfs , которое является расширением pyfilesystem2 для включения SFTP, должно помочь. ПРИМЕЧАНИЕ ЗДЕСЬ, на самом деле это не очень хорошо с opencv-python.

EDIT1:

из документов здесь вы можете увидеть, каким образом вы можете передать файл в VideoCapture.Open (). enter image description here

Редактирование кода для локального копирования файла и последующей передачи локального файла в openCV работает правильно. 

sftp.get('file.mp4', 'file.mp4')
sftp.close() # Also, close the sftp connection

cap = cv2.VideoCapture.open('file.mp4')

EDIT2:

Итак, монтирование файловой системы SFTP к локальной файловой системе с использованием ssfhs работает. Лучшим способом было бы смонтировать SFTP на уровне ОС, используя проверенные методы для этого. Ниже приведен пример кода Python, который делает все в Python, но обратите внимание, что предполагается, что ssfhs может правильно подключаться к хосту SFTP из командной строки. Я не объясняю эту часть здесь, поскольку есть превосходные разные учебные пособия для этого.

Обратите внимание, что здесь содержится только некоторая базовая проверка ошибок, поэтому я советую убедиться, что вы обнаружите все ошибки, которые могут появиться. Это доказательство концепции. 

import cv2
import os
import subprocess


g_remoteuser = 'USERNAME'
g_remotepassword = 'PASSWORD'
g_remotehost = 'HOSTIP'
g_remotepath = '/home/{remoteuser}/files'.format(remoteuser=g_remoteuser)
g_localuser = 'LOCAL_MACHINE_LINUX_USERNAME'
g_localmntpath = '/home/{localuser}/mnt/remotehost/'.format(localuser=g_localuser)
g_filename = 'file.mp4'


def check_if_path_exists(path):
    # check if the path exists, create the path if it doesn't
    if not os.path.exists(path):
        os.makedirs(path)


def mount(remoteuser, remotehost, remotepath, remotepassword, localmntpath):
    check_if_path_exists(localmntpath)
    if not check_if_mounted(localmntpath):
        subprocess.call([
            '''echo "{remotepassword}" | sshfs {remoteuser}@{remotehost}:{remotepath} {localmntpath} \
             -o password_stdin -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no -o auto_unmount -o allow_other'''.format(
                remoteuser=remoteuser,
                remotehost=remotehost,
                remotepath=remotepath,
                localmntpath=localmntpath,
                remotepassword=remotepassword
            )], shell=True)


def unmount(path):
    try:
        subprocess.call(['sudo umount -l {path}'.format(path=path)], shell=True)
    except Exception as e:
        print(e)


def check_if_mounted(path):
    # check if there's actually files. Hacky way to check if the remote host is already mounted.
    # will of course fail if there's no files in the remotehost
    from os import walk
    f = []
    for (dirpath, dirnames, filenames) in walk(path):
        f.extend(filenames)
        f.extend(dirnames)
        if dirnames or filenames or f:
            return True
        break
    return False


if check_if_mounted(g_localmntpath):
    unmount(g_localmntpath)

mount(g_remoteuser, g_remotehost, g_remotepath, g_remotepassword, g_localmntpath)


cap = cv2.VideoCapture()
cap.open(g_localmntpath + g_filename)


while True:
    _, frame = cap.read()
    print(frame)
    cv2.imshow('res', frame)
    key = cv2.waitKey(1)
    if key == 27:
        break

cap.release()
cv2.destroyAllWindows()
unmount(g_localmntpath)
...