Skip to content

Instantly share code, notes, and snippets.

View tori29umai0123's full-sized avatar
🏠

tori29umai tori29umai0123

🏠
View GitHub Profile
@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