/
stupin
/
AutoDriver
Обзор
Документация
Войти
/
stupin
/
AutoDriver
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
Dev/dataset/Maker.py
50 строк
2 KB
Stupin
init
25 июн 2025, 19:30
25 июн 2025, 19:30
a4828f6
Код
Авторство
О чём код?
from PIL import Image from matplotlib import pyplot as plt import random import math import json MAP_SIZE = 300 labels = [] for i in range(10000): if i % 100 == 0: print(i) info = [random.randint(0, 360)] info.extend([random.randint(0, MAP_SIZE - 30) for _ in range(8)]) MAP_OFFSET = MAP_SIZE / 128 labels.append([info[0] / 360]) labels[-1].extend([((round(info[_] / MAP_OFFSET)) + 6) / 128 for _ in range(1, 3)]) labels[-1].extend([((info[_] // MAP_OFFSET) + (4 if _ % 2 == 0 else 2)) / 128 for _ in range(3, 9)]) minimap = Image.open('dataset/resources/map.png') cursor = Image.open('dataset/resources/cursor.png') task = Image.open('dataset/resources/task.png') cursor = cursor.rotate(info[0], expand=True).resize((28, 28)) minimap.paste(cursor, (info[1], info[2]), cursor) minimap.paste(task, (info[3], info[4]), task) minimap.paste(task, (info[5], info[6]), task) minimap.paste(task, (info[7], info[8]), task) minimap = minimap.resize((128, 128)) # Generator tester #minimap.putpixel((int(labels[-1][1] * 128), int(labels[-1][2] * 128)), (255, 0, 0)) #minimap.putpixel((int(labels[-1][3] * 128), int(labels[-1][4] * 128)), (255, 0, 0)) #minimap.putpixel((int(labels[-1][5] * 128), int(labels[-1][6] * 128)), (255, 0, 0)) #minimap.putpixel((int(labels[-1][7] * 128), int(labels[-1][8] * 128)), (255, 0, 0)) minimap.save(f'dataset/quests/{i}.png') with open('dataset/quests/labels.json', 'w', encoding='utf-8') as f: f.write(json.dumps(labels, ensure_ascii=False)) ''' img = plt.imread('dataset/resources/background.png') fig, ax = plt.subplots() ax.imshow(img, extent=[0, 20, 0, 20]) ax.plot([13, 16, 14, 11, 10, 12, 18, 12, 15, 14, 17, 15, 13]) plt.show() '''