Receiver_black_and_white.py

import zmq
import pyaudio
from cv2 import VideoCapture
from util import SMOOTH,Filter,SHARPEN
import cv2
import numpy as np
import struct
import bz2
import time
import sys

ip = "127.0.0.1"
port = 8000

try:
    ip = sys.argv[1] 
    port = sys.argv[2]
except:
    print("Please include the server IP and port")
    exit()

context = zmq.Context()
SubscribeSocket = context.socket(zmq.SUB)
SubscribeSocket.connect("tcp://"+ip+":%s" % port) #Connect to server
SubscribeSocket.setsockopt(zmq.SUBSCRIBE, b"")

def AddFPS(frame1,string):
    cv2.putText(frame1,string,(0,25),cv2.FONT_HERSHEY_SIMPLEX,1,(255,0,0),2, cv2.LINE_AA)
    return frame1

frames = 0
fps = 30
timea = time.time() 

while True:
    
    compressed = SubscribeSocket.recv() #Receive packet
    buffer = bz2.decompress(compressed)
    shape = struct.unpack("HH",buffer[1:5])
    frameBytes = buffer[5:] #Get Frame Bytes
    frame = np.frombuffer(frameBytes,dtype = np.uint8) #Convert bytes to frame array
    #frame = np.unpackbits(np.asarray(frame,dtype=np.uint8))
    frame.shape = shape
    #frame = (frame*255)
    frame = cv2.resize(frame,(int(640*1.5),int(480*1.5))) #Upscale Frame 
    frame = Filter(frame,SMOOTH)
    frames += 1
    if(time.time() - timea > 5):
        timea = time.time()
        fps = frames//5
        frames = 0
    
    cv2.imshow("Video",AddFPS(frame,str(fps)+" FPS")) #Display the frame
    cv2.waitKey(1)