[Python] Image over TCP/IP through AXIDMA on FPGA and back

import time
import socket  # used for TCP/IP communication
import numpy as np

from PIL import Image

TCP_IP = '10.42.0.94'
TCP_PORT = 7

BUFFER_SIZE = 1000

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((TCP_IP, TCP_PORT))

print("Connected. Test started...")

im = Image.open('pivcuga.jpg')
im.show()

arr = np.array(im, dtype='u1').flatten()
package_count = len(arr) / BUFFER_SIZE
img_chunks = np.split(arr, package_count)

print("-- Image size: " + str(len(arr)) + " bytes, packet count: " + str(len(img_chunks)))

returned_img = []

print("-- Sent image size: " + str(len(arr)) + " bytes" )
for i in range(0, package_count):
    s.send(img_chunks[i].tostring())

    data = s.recv(BUFFER_SIZE)
    recv_data = np.fromstring(data, dtype='u1')
    returned_img.append(recv_data)

print("Done!")

img_arr = np.asarray(returned_img)
flatten_img = np.ndarray.flatten(img_arr)
img_arr = np.reshape(flatten_img, (1000, 1000, 3))

print("-- Returned len: " + str(len(img_arr)) + ", flatten len: "  + str(len(flatten_img)))
img = Image.fromarray(img_arr)
img.show()

s.close()