mirror of
https://github.com/kuhyx/WUT_Computer_Science.git
synced 2026-07-04 14:03:08 +02:00
Krzysztof kuhy Rudnicki
be962d0c35
git-subtree-dir: Programming/EARIN git-subtree-mainline:635e287095git-subtree-split:a09c96dd65
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
import tempfile
|
|
import os
|
|
|
|
import cv2
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
def main():
|
|
# define number of data points
|
|
N = 10
|
|
|
|
# define the visualization params
|
|
colors = np.random.rand(N)
|
|
|
|
with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as f:
|
|
# iterate over the optimization steps
|
|
for i in range(10):
|
|
# generate random 2D data - replace it with the results from your algorithm
|
|
x = np.random.rand(N)
|
|
y = np.random.rand(N)
|
|
|
|
# plot the data
|
|
plt.cla()
|
|
plt.figure()
|
|
plt.scatter(x, y, c=colors, alpha=0.5)
|
|
plt.xlim([0, 1])
|
|
plt.ylim([0, 1])
|
|
plt.savefig(f.name)
|
|
|
|
# read image
|
|
image = cv2.imread(f.name)
|
|
|
|
# show the image, provide window name first
|
|
cv2.imshow('visualization', image)
|
|
|
|
# add wait key. window waits until user presses a key and quits if the key is 'q'
|
|
if cv2.waitKey(0) == 113:
|
|
# and finally destroy/close all open windows
|
|
cv2.destroyAllWindows()
|
|
exit()
|
|
|
|
try:
|
|
f.close()
|
|
os.unlink(f.name)
|
|
except:
|
|
pass
|
|
|
|
cv2.destroyAllWindows()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|