-
Notifications
You must be signed in to change notification settings - Fork 0
/
z.py
64 lines (59 loc) · 2.04 KB
/
z.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#!ipython
# -*- coding:utf-8 -*-
from PIL import Image, ImageDraw
import numpy as np
from npcolorconvert import npjch2rgb, nprgb2jch
import matplotlib.pyplot as plt
import random
import string
def show_jch_rgb(jch, rgb, figsize=(8, 6)):
fig = plt.figure(figsize=figsize)
w, h = figsize
length = len(jch)
xtick = np.arange(length)
if length > 100:
xtick = np.arange(length//(length//100))*(length//100)
plt.subplots_adjust(0.5/w, 0.6/h, 1-0.3/w, 1-0.3/h, 0, 0.6/h)
p1 = plt.subplot(211)
p1.set_xticks(xtick)
p1.set_xticklabels(xtick, rotation=-90)
p1.grid(True)
p1.plot(jch[:, 0], '|-', color=(0.1, 0.1, 0.1))
p1.plot(jch[:, 1], '|-', color=(0.7, 0.7, 0.7))
p1.plot(jch[:, 2], '|-', color=(0.6, 0.4, 0.8))
p2 = plt.subplot(212)
p2.set_xticks(xtick)
p2.set_xticklabels(xtick, rotation=-90)
p2.grid(True)
p2.plot(rgb[:, 0], '|-', color=(0.6, 0.1, 0.1))
p2.plot(rgb[:, 1], '|-', color=(0.1, 0.7, 0.2))
p2.plot(rgb[:, 2], '|-', color=(0.1, 0.3, 0.7))
filename = '/tmp/' + \
''.join(random.sample(string.ascii_letters+string.digits, 8))+'.png'
plt.savefig(filename)
plt.close()
im = Image.open(filename)
im.show()
def plot_jch_rgb(jch, rgb, figsize=(8, 6)):
fig = plt.figure(figsize=figsize)
w, h = figsize
length = len(jch)
xtick = np.arange(length)
if length >= 100:
xtick = np.arange(length//(length//100))*(length//100)
plt.subplots_adjust(0.5/w, 0.6/h, 1-0.3/w, 1-0.3/h, 0, 0.6/h)
p1 = plt.subplot(211)
p1.set_xticks(xtick)
p1.set_xticklabels(xtick, rotation=-90)
p1.grid(True)
p1.plot(jch[:, 0], '|-', color=(0.1, 0.1, 0.1))
p1.plot(jch[:, 1], '|-', color=(0.7, 0.7, 0.7))
p1.plot(jch[:, 2], '|-', color=(0.6, 0.4, 0.8))
p2 = plt.subplot(212)
p2.set_xticks(xtick)
p2.set_xticklabels(xtick, rotation=-90)
p2.grid(True)
p2.plot(rgb[:, 0], '|-', color=(0.6, 0.1, 0.1))
p2.plot(rgb[:, 1], '|-', color=(0.1, 0.7, 0.2))
p2.plot(rgb[:, 2], '|-', color=(0.1, 0.3, 0.7))
plt.show()