47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
import requests
|
|
import json
|
|
import time
|
|
import os
|
|
API_URL = "http://127.0.0.1:8008"
|
|
|
|
# 테스트할 이미지 번호 (1~7)
|
|
img_num = 5 # 필요시 1~7로 변경
|
|
img_path = os.path.join(os.path.dirname(__file__), "modules","img",f"{img_num}.jpg")
|
|
|
|
# 저장할 결과 이미지 경로
|
|
output_image_path = os.path.join(os.path.dirname(img_path), f"result_{img_num}.png")
|
|
|
|
# # /translate_me 테스트
|
|
# translate_payload = {
|
|
# "local_image_path": img_path,
|
|
# "unwanted_words": {"테스트": "치환값", "삭제": "이미지삭제"},
|
|
# "method": "lama",
|
|
# "index": 0,
|
|
# "delay": 1.0,
|
|
# "output_image_path": output_image_path,
|
|
# "toggle_states": {"ocr": True, "watermark_text": "테스트워터마크", "remove_background_white": True}
|
|
# }
|
|
|
|
# print("/translate_me 요청 시작...")
|
|
# start_time = time.time()
|
|
# resp = requests.post(f"{API_URL}/translate_me", json=translate_payload, timeout=40)
|
|
# elapsed = time.time() - start_time
|
|
# print(f"/translate_me 결과 (소요 시간: {elapsed:.2f}초):")
|
|
# print(json.dumps(resp.json(), ensure_ascii=False, indent=2))
|
|
|
|
# /remove_bg_me 테스트
|
|
remove_bg_payload = {
|
|
"local_image_path": img_path,
|
|
"index": 0,
|
|
"output_image_path": output_image_path,
|
|
"toggle_states": {"remove_background_white": True}
|
|
}
|
|
|
|
print("\n/remove_bg_me 요청 시작...")
|
|
start_time = time.time()
|
|
resp2 = requests.post(f"{API_URL}/remove_bg_me", json=remove_bg_payload, timeout=40)
|
|
elapsed2 = time.time() - start_time
|
|
print(f"/remove_bg_me 결과 (소요 시간: {elapsed2:.2f}초):")
|
|
print(json.dumps(resp2.json(), ensure_ascii=False, indent=2))
|
|
|
|
# 이미지 번호만 바꿔가며 테스트 가능 |