Skip to content

Instantly share code, notes, and snippets.

View tori29umai0123's full-sized avatar
🏠

tori29umai tori29umai0123

🏠
View GitHub Profile
@tori29umai0123
tori29umai0123 / rag_stable_diffusion_prompt.py
Last active July 9, 2024 12:36
rag_stable_diffusion_prompt.py
import csv
from pathlib import Path
import re
from sentence_transformers import SentenceTransformer
import faiss
from llama_cpp import Llama
import numpy as np
@tori29umai0123
tori29umai0123 / danbooru_tags.csv
Created July 9, 2024 10:21
danbooru_tags.csv
We can't make this file beautiful and searchable because it's too large.
1girl,4114588
solo,3426446
highres,3008413
long_hair,2898315
commentary_request,2610959
breasts,2252741
looking_at_viewer,2017032
blush,1987494
smile,1847264
short_hair,1530601
import csv
from pathlib import Path
import re
from langchain_core.prompts import ChatPromptTemplate
from langchain.chains import create_retrieval_chain
from langchain.chains.combine_documents import create_stuff_documents_chain
from langchain_community.vectorstores import FAISS
from langchain_community.llms import LlamaCpp
from langchain.prompts import PromptTemplate
from langchain_core.documents import Document
import os
from concurrent.futures import ThreadPoolExecutor
from PIL import Image, ImageFilter
from PIL import Image, ImageFilter
def apply_blur_to_image(image_path, output_path, blur_radius):
"""指定された画像にブラー効果を適用し、指定されたパスに保存する"""
try:
image = Image.open(image_path)
#線画あり画像から、線画を取り除くスクリプト。線画を取り除きたい画像(image_dir)があれば動く。線画入力不要。
from PIL import Image, ImageFilter, ImageOps
from collections import defaultdict
from skimage import color as sk_color
from tqdm import tqdm
from skimage.color import deltaE_ciede2000, rgb2lab
import cv2
import numpy as np
import os
@tori29umai0123
tori29umai0123 / ContentSafetyAnalyzer.py
Last active June 14, 2024 02:36
ContentSafetyAnalyzer.py
import csv
import os
from pathlib import Path
import cv2
import numpy as np
from PIL import Image
import onnxruntime as ort
from huggingface_hub import hf_hub_download
@tori29umai0123
tori29umai0123 / nsfw_filter_with_tagger.py
Last active June 5, 2024 01:47
nsfw_filter_with_tagger.py
import csv
import glob
import os
from pathlib import Path
import cv2
import numpy as np
import torch
from PIL import Image
from tqdm import tqdm
@tori29umai0123
tori29umai0123 / HF_upload_sdxl_gen_img.py
Last active May 28, 2024 23:20
HF_upload_sdxl_gen_img.py
import os
import sys
import wget
import zipfile
import threading
import re
from time import sleep
import shutil
from huggingface_hub import HfApi, Repository, upload_file
import gen_img
#https://github.com/Kazuhito00/XDoG-OpenCV-Sample
import os
import cv2
import numpy as np
def DoG_filter(image, kernel_size=0, sigma=1.0, k_sigma=2.0, gamma=1.5):
g1 = cv2.GaussianBlur(image, (kernel_size, kernel_size), sigma)
g2 = cv2.GaussianBlur(image, (kernel_size, kernel_size), sigma * k_sigma)
return g1 - gamma * g2
@tori29umai0123
tori29umai0123 / nsfw_filter.py
Last active June 11, 2024 13:02
nsfw_filter.py
import argparse
import csv
import glob
import os
from pathlib import Path
import cv2
import numpy as np
import torch
from PIL import Image