File size: 1,327 Bytes
af2d09d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
from flask import Flask, jsonify, request, send_file
from flask_cors import CORS
from PIL import Image
import io
import numpy as np
import cv2
# Initialize the Flask app
myapp = Flask(__name__)
CORS(myapp) # Enable CORS if needed
@myapp.route('/')
def home():
return "Welcome to the Image Denoiser!" # Basic home response
@myapp.route('/denoise', methods=['POST'])
def denoise_image():
if 'image' not in request.files:
return jsonify({"error": "No image provided"}), 400
input_image = request.files['image'].read() # Read the uploaded image
img = Image.open(io.BytesIO(input_image))
# Convert to OpenCV format
img_cv = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
# Apply fastNlMeansDenoising
denoised_img = cv2.fastNlMeansDenoising(img_cv, None, h=10, templateWindowSize=7, searchWindowSize=21)
# Convert back to PIL format
denoised_img_pil = Image.fromarray(cv2.cvtColor(denoised_img, cv2.COLOR_BGR2RGB))
# Save to bytes
img_byte_arr = io.BytesIO()
denoised_img_pil.save(img_byte_arr, format='PNG')
img_byte_arr.seek(0)
return send_file(img_byte_arr, mimetype='image/png')
# Add this block to make sure your app runs when called
if __name__ == "__main__":
myapp.run(host='0.0.0.0', port=7860) # Run directly if needed for testing |