Upload 11 files

Browse files

Files changed (12) hide show

.gitattributes +1 -0
image_classifier.ipynb +0 -0
image_classify.keras +3 -0
main.py +64 -0
model.py +22 -0
requirements.txt +0 -0
static/GantMan_0B31E09D-12C5-465A-9711-2AA32E31DB61.jpg +0 -0
templates/index.html +54 -0
web scrap scripts/ai_scrap.py +64 -0
web scrap scripts/ai_script2.py +82 -0
web scrap scripts/rename.py +17 -0
web scrap scripts/resize.py +16 -0

.gitattributes CHANGED Viewed

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text

 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+image_classify.keras filter=lfs diff=lfs merge=lfs -text

image_classifier.ipynb ADDED Viewed

The diff for this file is too large to render. See raw diff

image_classify.keras ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:ae3afde9171a7b7f2fbcbee9cb2c3e7ac183308190a5f29c6f1d826e3275d745
+size 574861816

main.py ADDED Viewed

	@@ -0,0 +1,64 @@

+import os
+from flask import Flask, flash, request, redirect, url_for,render_template
+from werkzeug.utils import secure_filename
+from model import check
+from PIL import Image
+UPLOAD_FOLDER = 'static'
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
+app = Flask(__name__)
+app.secret_key = "secret key"
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+def allowed_file(filename):
+    return '.' in filename and \
+           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+@app.route("/")
+def home():
+    static_folder = app.static_folder
+    for filename in os.listdir(static_folder):
+        file_path = os.path.join(static_folder, filename)
+        if os.path.isfile(file_path):
+            os.remove(file_path)
+    return render_template("index.html")
+@app.route("/edit", methods = {"GET","POST"})
+def edit():
+    if request.method == "POST":
+        static_folder = app.static_folder
+        for filename in os.listdir(static_folder):
+            file_path = os.path.join(static_folder, filename)
+            if os.path.isfile(file_path):
+                os.remove(file_path)
+        if 'file' not in request.files:
+            flash('No file part')
+            return "error"
+        file = request.files['file']
+        # If the user does not select a file, the browser submits an
+        # empty file without a filename.
+        if file.filename == '':
+            flash('No selected file')
+            return render_template("index.html")
+        if file and allowed_file(file.filename):
+            filename = secure_filename(file.filename)
+            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+            image = Image.open(file)
+            pre = check(image)
+            if(pre >= 0.55):
+                flash('REAL IMAGE')
+            else:
+                flash('AI GENERATED IMAGE')
+            return render_template("index.html", filename = filename)
+        else:
+            flash('Allowed file types are: png, jpg, jpeg')
+    return render_template("index.html")
+@app.route("/display/<filename>")
+def display_image(filename):
+    return redirect(url_for('static', filename = filename), code = 301)
+app.run(debug=True)

model.py ADDED Viewed

	@@ -0,0 +1,22 @@

+from PIL import Image
+import numpy as np
+from tensorflow import keras
+import os
+def check(image):
+    # Load the image
+    # Preprocess the image
+    image = image.resize((300, 300))  # Resize the image to the desired dimensions
+    image = np.array(image)  # Convert the image to a numpy array
+    image = image.astype('float32') / 255.0  # Normalize pixel values between 0 and 1
+    # Expand dimensions and create a batch
+    image = np.expand_dims(image, axis=0)
+    model = keras.models.load_model('.\\image_classify.keras')
+    # Make predictions
+    predictions = model.predict(image)
+    return predictions

requirements.txt ADDED Viewed

Binary file (2.5 kB). View file

static/GantMan_0B31E09D-12C5-465A-9711-2AA32E31DB61.jpg ADDED Viewed

templates/index.html ADDED Viewed

	@@ -0,0 +1,54 @@

+<!doctype html>
+<html lang="en" data-bs-theme="dark">
+<head>
+  <meta charset="utf-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1">
+  <title>Image Classifier</title>
+  <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/css/bootstrap.min.css" rel="stylesheet"
+    integrity="sha384-9ndCyUaIbzAi2FUVXJi0CjmCapSmO7SnpJef0486qhLnuZ2cdeRhO02iuK6FUUVM" crossorigin="anonymous">
+</head>
+<body>
+  <center>
+    <div class="container my-4">
+      <h1 class="fs-2 text"> Image Classifier - Is your image AI generated?</h1>
+      <form action="/edit" method="post" enctype="multipart/form-data">
+        <div class="mb-3">
+          <label for="formFile" class="form-label">Select Image to classify</label>
+          <input class="form-control" type="file" name="file" id="formFile">
+        </div>
+        <button type="submit" class="btn btn-success">Submit</button>
+      </form>
+    </div>
+    <div>
+      {% if filename %}
+      <div style="padding: 20px;">
+        <img src="{{url_for('display_image', filename=filename)}} " style="height: 30%; width: 30%;">
+      </div>
+      {% endif %}
+      <p>
+        <h4>
+          {% with messages = get_flashed_messages() %}
+        {% if messages %}
+      <ul class=flashes>
+        {% for message in messages %}
+        <li>{{ message }}</li>
+        {% endfor %}
+      </ul>
+      {% endif %}
+      {% endwith %}
+        </h4>
+      </p>
+    </div>
+  </center>
+  <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/js/bootstrap.bundle.min.js"
+    integrity="sha384-geWF76RCwLtnZ8qwWowPQNguL3RmwHVBC9FhGdlKrxdiJJigb/j/68SIy3Te4Bkz"
+    crossorigin="anonymous"></script>
+</body>
+</html>

web scrap scripts/ai_scrap.py ADDED Viewed

	@@ -0,0 +1,64 @@

+import bs4
+from PIL import Image
+from selenium.webdriver.chrome.service import Service
+from selenium import webdriver
+import requests
+import os
+import time
+urlW = "https://impossibleimages.ai/images/"
+PATH = r"C:\Users\anujg\Downloads\Programs\chromedriver_win32\chromedriver.exe"
+service = Service(PATH)
+options = webdriver.ChromeOptions()
+options.add_argument('--no-sandbox')
+options.add_argument('--headless')
+options.add_argument('--disable-gpu')
+options.add_argument('--disable-dev-shm-usage')
+options.add_argument("--window-size=1920,1080")
+driver = webdriver.Chrome(service=service, options=options)
+# driver.get(urlW)
+# os.makedirs('ai-generated-images')
+# time.sleep(2)
+def dloadimg(url, name):
+    # write image to file
+    reponse = requests.get(url)
+    if reponse.status_code==200:
+        with open(os.path.join('ai2',name+".jpg"), 'wb') as file:
+            file.write(reponse.content)
+def section_length(driver):
+    fig_xpath = '//*[@id="page"]/main/section/div/div/div[2]/article'
+    figures = driver.find_elements(by="xpath", value=fig_xpath)
+    return len(figures)
+def dload(urlW,name):
+    driver.get(urlW)
+    driver.find_element(by="id", value="CybotCookiebotDialogBodyLevelButtonAccept").click()
+    for j in range(30):
+        j = j + 1
+        time.sleep(15)
+        driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
+    time.sleep(5)
+    num_img = section_length(driver)
+    print(num_img)
+    for i in range(1, num_img+1):
+        ImageXPath = """//*[@id="page"]/main/section/div/div/div[2]/article[%s]/div/img"""%(i)
+        # ImageElement = driver.find_element_by_xpath(ImageXPath)
+        ImageElement = driver.find_element(by="xpath", value=ImageXPath)
+        ImageURL = ImageElement.get_attribute("src")
+        print(ImageURL)
+        try:
+            dloadimg(ImageURL, name+str(i))
+            print("Downloaded element %s out of %s total." % (i, num_img))
+        except:
+            print("Couldn't download an image %s, continuing downloading the next one"%(i))
+dload(urlW,"AI_Img_imp")
+# time.sleep(100)
+driver.close()

web scrap scripts/ai_script2.py ADDED Viewed

	@@ -0,0 +1,82 @@

+import bs4
+from PIL import Image
+from selenium.webdriver.chrome.service import Service
+from selenium import webdriver
+import requests
+import os
+import time
+PATH = r"C:\Users\anujg\Downloads\Programs\chromedriver_win32\chromedriver.exe"
+service = Service(PATH)
+options = webdriver.ChromeOptions()
+options.add_argument('--no-sandbox')
+options.add_argument('--headless')
+options.add_argument('--disable-gpu')
+options.add_argument('--disable-dev-shm-usage')
+options.add_argument("--window-size=1920,10800")
+# Made window size so much that 50 images can fit in one window so that all images will be loaded
+driver = webdriver.Chrome(service=service, options=options)
+# driver.get(urlW)
+# os.makedirs('ai-generated-images')
+def dloadimg(url, name):
+    # write image to file
+    reponse = requests.get(url)
+    if reponse.status_code==200:
+        with open(os.path.join('resized',name+".jpg"), 'wb') as file:
+            # # file = file.resize((400, 400))
+            file.write(reponse.content)
+            # try:
+            #     image = Image.open(reponse.content)
+            #     image = image.resize((400, 400))
+            #     image.save(file)
+            # except Exception as e:
+            #     print(e)
+def section_length(driver):
+    fig_xpath = '//*[@id="main"]/div[3]/div/div[2]/section/figure'
+    figures = driver.find_elements(by="xpath", value=fig_xpath)
+    return len(figures)
+def dload(urlW,name,pc):
+        driver.get(urlW)
+        num_img = section_length(driver)
+        print(num_img)
+        for i in range(1, num_img+1):
+            ImageXPath = """//*[@id="main"]/div[3]/div/div[2]/section/figure[%s]/div/a/img"""%(i)
+            # ImageElement = driver.find_element_by_xpath(ImageXPath)
+            timeStarted = time.time()
+            while True:
+                try:
+                    ImageElement = driver.find_element(by="xpath", value=ImageXPath)
+                    ImageURL = ImageElement.get_attribute("src")
+                    if ImageURL == "https://freepik.cdnpk.net/img/1px.png":
+                        currentTime = time.time()
+                        if currentTime - timeStarted > 10:
+                            print("Timeout! moving to next image")
+                            break
+                    elif ImageURL != "https://freepik.cdnpk.net/img/1px.png":
+                        dloadimg(ImageURL, name+"_pg_no."+str(pc)+"img"+str(i))
+                        print("Downloaded image no. %s of page no. %s." %(i,pc))
+                        break
+                    else:
+                        break
+                except:
+                    print("Couldn't download an image %s, continuing downloading the next one"%(i))
+                    break
+pgcnt  = 1
+while pgcnt<23:
+    urlW = "https://www.freepik.com/photos/ai-generated/%s"%(pgcnt)
+    dload(urlW,"AI_Img",pgcnt)
+    pgcnt = pgcnt + 1
+driver.close()

web scrap scripts/rename.py ADDED Viewed

	@@ -0,0 +1,17 @@

+import os
+import PIL.Image
+def rename_imgs(org_folder, renamed_folder, name):
+    i = 1
+    for file in os.listdir(org_folder):
+        img = PIL.Image.open(os.path.join(org_folder, file))
+        # img = img.resize((300,300))
+        # if img.mode == "RGBA":
+        #     img = img.convert("RGB")
+        img.save(os.path.join(renamed_folder, name+str(i)+".jpg"))
+        i += 1
+if __name__ == "__main__":
+    rename_imgs("real_small", "Real", "real_train")
+    rename_imgs("ai_small", "AI generated", "ai_train")

web scrap scripts/resize.py ADDED Viewed

	@@ -0,0 +1,16 @@

+import os
+import PIL.Image
+def resize_imgs(org_folder, resized_folder, name):
+    i = 1
+    for file in os.listdir(org_folder):
+        img = PIL.Image.open(os.path.join(org_folder, file))
+        img = img.resize((300,300))
+        if img.mode == "RGBA":
+            img = img.convert("RGB")
+        img.save(os.path.join(resized_folder, name+str(i)+".jpg"))
+        i += 1
+if __name__ == "__main__":
+    resize_imgs("ai_imgs", "ai_resized", "ai_imgs_")
+    resize_imgs("real_imgs", "real_resized", "real_imgs_")