multiprocessing_pipe_numpy

multiprocessing_pipe_numpy.py

from multiprocessing import Process, Pipe
import time
import numpy as np
  
def send_to_pipe(q, mylist): 
    # function to put elements into Pipe
    for num in mylist: 
        q.send(np.random.random((416, 416, 3)))
  
def pick_from_pipe(q, mylist): 
    # function to print pipe elements 
    print("Pipe elements:")
    t1 = time.time()
    for num in mylist:
        result = q.recv() 
        #print(result)
    print("Pipe is now empty!", time.time()-t1) 
  
if __name__ == "__main__": 
    # input list 
    mylist = [i for i in range(100)] 
  
    # creating multiprocessing Pipe 
    parent_conn, child_conn = Pipe() 
  
    # creating new processes 
    p1 = Process(target=send_to_pipe, args=(parent_conn, mylist)) 
    p2 = Process(target=pick_from_pipe, args=(child_conn, mylist)) 
  
    # running process p1 and p2
    p1.start()
    p2.start() 
  
    # if we need waiting p1 and p2 process to finish their job
    #p1.join() 
    #p2.join()