Skip to content

Instantly share code, notes, and snippets.

#
# フォルダ以下のPNGファイルのメタデータを全部読み込んでHTMLに書き出す
# Stable Diffusion で参考になるプロンプトを探すために作成
#
require 'chunky_png'
def template_html
return <<'EOS'
<!DOCTYPE html>
#
# PNGファイルのメタデータを全部読み込んでメモ帳に表示
#
require 'chunky_png'
require 'securerandom'
tempdir = 'C:\SD\pnglog'
if (ARGV.count < 1)
puts "please set parameter for file"
exit
# 渡ってきた複数ファイルのうち、拡張子がsafetensorsのファイルを探し、
# それ以外の画像ファイルをすべて同じファイル名に変える。
# CivitaiからダウンロードしてきたLoRAに対して使う。
require 'fileutils'
def rename_files_to_same_name
imagefiles = []
safefilename = ""
ARGV.each do |file|
import os
from glob import glob
from skimage.metrics import structural_similarity as ssim
import cv2
import sys
from pathlib import Path
def calculate_ssim(image_path1, image_path2):
"""2つの画像間のSSIM(構造的類似性)を計算する"""
img1 = cv2.imread(image_path1, cv2.IMREAD_GRAYSCALE)
# Civitaiのinfoファイルを使ってLoraを分類するコード
# プレビュー画像は取り直しなので注意
require 'json'
require 'fileutils'
# === 設定 ===
TARGET_DIR = 'C:\SD\webui\models\Lora\nofolder' # 対象のフォルダに書き換えてください
DEST_BASE_DIR = 'C:\SD\webui\models\Lora\copied' # 分類結果を保存するフォルダ