server.py

import socket
import sys

import cv2
import numpy as np

import struct

HOST = None               # Symbolic name meaning all available interfaces
PORT = 7200              # Arbitrary non-privileged port
s = None
FRAME_WIDTH = 640
FRAME_HEIGHT = 480
CHANNEL=3


def socketToNumpy(cameraFeed, sockData):
    k=3
    j=cameraFeed.shape[1]
    i=cameraFeed.shape[0]
    sockData = np.fromstring(sockData, np.uint8)
    cameraFeed = np.tile(sockData, 1).reshape((i,j,k))

    return cameraFeed

for res in socket.getaddrinfo(HOST, PORT, socket.AF_UNSPEC,
                              socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
    af, socktype, proto, canonname, sa = res
    try:
        s = socket.socket(af, socktype, proto)
    except socket.error as msg:
        s = None
        continue
    try:
        s.bind(sa)
        s.listen(1)
    except socket.error as msg:
        s.close()
        s = None
        continue
    break
if s is None:
    print 'could not open socket'
    sys.exit(1)
conn, addr = s.accept()
print 'Connected by', addr

running = True
while running:
    i, ptr = (0,0)
    shape = (FRAME_HEIGHT, FRAME_WIDTH, CHANNEL)
    cameraFeed = np.zeros(shape, np.uint8)
    imgSize = cameraFeed.size
    sockData = b''
    result = True

    while imgSize:
        nbytes=conn.recv(imgSize)
        if not nbytes: break; result = False
        sockData+=nbytes
        imgSize-=len(nbytes)

    if result:
        cameraFeed = socketToNumpy(cameraFeed, sockData)

        # Create a window for display.
        cv2.namedWindow("server");
        cv2.imshow("server", cameraFeed)
        key = cv2.waitKey(30)
        running = key

        # esc
        if key==27:
            running =False
    else : running =False
conn.close()