19 lines
557 B
Python
19 lines
557 B
Python
import numpy as np
|
|
from matplotlib.colors import rgb_to_hsv
|
|
|
|
from check import plot_preds
|
|
from utils import extract_colors
|
|
|
|
if __name__ == "__main__":
|
|
rgb_tensor, _ = extract_colors()
|
|
xkcd_rgb = rgb_tensor.numpy()
|
|
# xkcd_rgb = np.random.rand(1000, 3)
|
|
xkcd_hsv = rgb_to_hsv(xkcd_rgb)
|
|
plot_preds(
|
|
xkcd_hsv[:, 0], xkcd_rgb, fname="hsv", roll=True, dpi=300, figsize=(6, 6)
|
|
)
|
|
rgb = np.vstack([np.eye(3), np.eye(3) + np.eye(3)[:, [1, 2, 0]]])
|
|
print("Pure RGB in Hue-Space:")
|
|
print(rgb)
|
|
print(rgb_to_hsv(rgb)[:, 0])
|