askill
/
r_place
mirror of https://github.com/Askill/r_place.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added break off condition

This commit is contained in:
Askill 2022-09-25 16:43:09 +02:00
parent b82ebe2809
commit 370e30652c
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
client/clients.py
View File

@ -66,10 +66,11 @@ async def sender(target, img):
start_x = random.randint(0, 900) start_x = random.randint(0, 900)
start_y = random.randint(0, 900) start_y = random.randint(0, 900)
max_w, max_h, _ = img.shape max_w, max_h, _ = img.shape
i = 0
async for websocket in websockets.connect(target + "/set"): async for websocket in websockets.connect(target + "/set", timeout=10):
try: try:
for _ in range(int(max_h*max_w*1.3)): while i < max_h*max_w*1.5:
i+=1
rx = random.randint(0, max_w - 1) rx = random.randint(0, max_w - 1)
ry = random.randint(0, max_h - 1) ry = random.randint(0, max_h - 1)
if rx + start_x >= 1000 or ry + start_y >= 1000: if rx + start_x >= 1000 or ry + start_y >= 1000:
@ -111,6 +112,7 @@ def rescale(max_dimension, img):
async def main(target): async def main(target):
images_folder_path = "./images" images_folder_path = "./images"
while True: while True:
print("start cycle")
image_paths = os.listdir(images_folder_path) image_paths = os.listdir(images_folder_path)
images = [ images = [
np.array(rescale(random.randint(100, 400), Image.open(f"{images_folder_path}/{image_path}"))) np.array(rescale(random.randint(100, 400), Image.open(f"{images_folder_path}/{image_path}")))
@ -127,4 +129,4 @@ def asyncMain(x, target):
if __name__ == "__main__": if __name__ == "__main__":
# with Pool(12) as p: # with Pool(12) as p:
# print(p.map(asyncMain, [() for _ in range(12)])) # print(p.map(asyncMain, [() for _ in range(12)]))
asyncMain(0, target="ws://localhost:8080") asyncMain(0, target=os.getenv("TARGET", "ws://venus:8080"))