fix image gen

This commit is contained in:
Michael Pilosov, PhD 2024-01-26 03:02:33 +00:00
parent a02a662b6f
commit 50628e594a
4 changed files with 8 additions and 3 deletions

View File

@ -83,13 +83,15 @@ def plot_preds(
theta[i],
height=1,
width=width,
edgecolor="none",
edgecolor=colors[i],
linewidth=0.25,
# facecolor=[rgb_values[i][1]]*3,
# facecolor=rgb_values[i],
facecolor=colors[i],
bottom=0.0,
zorder=1,
alpha=1,
align='edge',
)
ax.set_xticks([])
@ -126,7 +128,7 @@ if __name__ == "__main__":
for v in versions:
name = f"out/v{v}"
# ckpt = f"/teamspace/jobs/{name}/work/colors/lightning_logs/version_2/checkpoints/epoch=999-step=8000.ckpt"
ckpt_path = f"/teamspace/studios/this_studio/colors/lightning_logs/version_{v}/checkpoints/*.ckpt"
ckpt_path = f"/teamspace/studios/colors-refactor-secondary/colors/lightning_logs/version_{v}/checkpoints/*.ckpt"
ckpt = glob.glob(ckpt_path)
if len(ckpt) > 0:
ckpt = ckpt[-1]

BIN
hsv.png

Binary file not shown.

Before

Width:  |  Height:  |  Size: 2.7 MiB

After

Width:  |  Height:  |  Size: 2.7 MiB

2
hsv.py
View File

@ -10,7 +10,7 @@ if __name__ == "__main__":
# xkcd_rgb = np.random.rand(1000, 3)
xkcd_hsv = rgb_to_hsv(xkcd_rgb)
plot_preds(
xkcd_hsv[:, 0], xkcd_rgb, fname="hsv", roll=False, dpi=300, figsize=(6, 6)
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:")

View File

@ -9,6 +9,9 @@ test:
search:
python search.py
hsv:
python hsv.py
animate:
ffmpeg -i lightning_logs/version_258/e%04d.png \
-c:v libx264 \