Upload 55 files

.gitattributes

@@ -2,27 +2,20 @@
 *.arrow filter=lfs diff=lfs merge=lfs -text
 *.bin filter=lfs diff=lfs merge=lfs -text
 *.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
 *.ftz filter=lfs diff=lfs merge=lfs -text
 *.gz filter=lfs diff=lfs merge=lfs -text
 *.h5 filter=lfs diff=lfs merge=lfs -text
 *.joblib filter=lfs diff=lfs merge=lfs -text
 *.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
 *.model filter=lfs diff=lfs merge=lfs -text
 *.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
 *.onnx filter=lfs diff=lfs merge=lfs -text
 *.ot filter=lfs diff=lfs merge=lfs -text
 *.parquet filter=lfs diff=lfs merge=lfs -text
 *.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
 *.pt filter=lfs diff=lfs merge=lfs -text
 *.pth filter=lfs diff=lfs merge=lfs -text
 *.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
 saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.tar.* filter=lfs diff=lfs merge=lfs -text
 *.tflite filter=lfs diff=lfs merge=lfs -text
@@ -30,8 +23,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.wasm filter=lfs diff=lfs merge=lfs -text
 *.xz filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-Background_Removal_and_Change/remove_bg/messi.jpg_4.png filter=lfs diff=lfs merge=lfs -text
-Background_Removal_and_Change/remove_bg/messi.jpg_5.png filter=lfs diff=lfs merge=lfs -text
-Background_Removal_and_Change/remove_bg/messi.jpg_6.png filter=lfs diff=lfs merge=lfs -text
+remove_bg/messi.jpg_4.png filter=lfs diff=lfs merge=lfs -text
+remove_bg/messi.jpg_5.png filter=lfs diff=lfs merge=lfs -text
+remove_bg/messi.jpg_6.png filter=lfs diff=lfs merge=lfs -text

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/Background_Removal_and_Change.iml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/venvme" />
+    </content>
+    <orderEntry type="jdk" jdkName="Python 3.9 (pythonProject)" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/deployment.xml

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="PublishConfigData" serverName="HAHALOLO_SERVER" remoteFilesAllowedToDisappearOnAutoupload="false">
+    <serverData>
+      <paths name="HAHALOLO_SERVER">
+        <serverdata>
+          <mappings>
+            <mapping deploy="/text_censorship" local="$USER_HOME$/PycharmProjects/TextCensorship" web="/" />
+          </mappings>
+        </serverdata>
+      </paths>
+    </serverData>
+  </component>
+</project>

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,27 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="14">
+            <item index="0" class="java.lang.String" itemvalue="scipy" />
+            <item index="1" class="java.lang.String" itemvalue="opencv-python" />
+            <item index="2" class="java.lang.String" itemvalue="Pillow" />
+            <item index="3" class="java.lang.String" itemvalue="pandas" />
+            <item index="4" class="java.lang.String" itemvalue="datetime" />
+            <item index="5" class="java.lang.String" itemvalue="tensorflow" />
+            <item index="6" class="java.lang.String" itemvalue="retina-face" />
+            <item index="7" class="java.lang.String" itemvalue="gdown" />
+            <item index="8" class="java.lang.String" itemvalue="matplotlib" />
+            <item index="9" class="java.lang.String" itemvalue="deepface" />
+            <item index="10" class="java.lang.String" itemvalue="cmake" />
+            <item index="11" class="java.lang.String" itemvalue="keras" />
+            <item index="12" class="java.lang.String" itemvalue="requests" />
+            <item index="13" class="java.lang.String" itemvalue="numpy" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9 (pythonProject)" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/Background_Removal_and_Change.iml" filepath="$PROJECT_DIR$/.idea/Background_Removal_and_Change.iml" />
+    </modules>
+  </component>
+</project>

.idea/sshConfigs.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="SshConfigs">
+    <configs>
+      <sshConfig authType="PASSWORD" host="10.10.15.87" id="8b67aa76-bd20-4263-bc45-270bf4ea8b6e" port="22" nameFormat="DESCRIPTIVE" username="root" useOpenSSHConfig="true" />
+    </configs>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$/Background_removal_and_change" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$/Remove-Photo-Background-using-TensorFlow" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$/yolov5" vcs="Git" />
+  </component>
+</project>

.idea/webServers.xml

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="WebServers">
+    <option name="servers">
+      <webServer id="7c3f60f5-5354-4ef4-a283-182d579b3fe7" name="HAHALOLO_SERVER">
+        <fileTransfer rootFolder="/storage/test" accessType="SFTP" host="10.10.15.87" port="22" sshConfigId="8b67aa76-bd20-4263-bc45-270bf4ea8b6e" sshConfig="[email protected]:22 password">
+          <advancedOptions>
+            <advancedOptions dataProtectionLevel="Private" keepAliveTimeout="0" passiveMode="true" shareSSLContext="true" isUseRsync="true" />
+          </advancedOptions>
+        </fileTransfer>
+      </webServer>
+    </option>
+  </component>
+</project>

Procfile

@@ -0,0 +1 @@
+web: sh setup.sh && streamlit run app.py

README.md

@@ -1,12 +1,9 @@
 ---
-title: Background Swapping
-emoji: 🌖
-colorFrom: red
+title: ABC- Automatic Background Change
+emoji: ⚡
+colorFrom: yellow
 colorTo: gray
 sdk: streamlit
-sdk_version: 1.17.0
 app_file: app.py
 pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---

app.py

@@ -0,0 +1,94 @@
+from libs import *
+from utils_func import create_dir, main_processing
+
+
+
+create_dir("tempDir")
+
+
+def load_image(image_file):
+	img = Image.open(image_file)
+	return img
+
+
+def streamlit_app():
+    detection_model_path = "weight_files/clothes_detection_model.pt"
+    background_model_path = "weight_files/model.h5"
+    save_path = ""
+    image_file = None
+    st.title("""WELCOME TO MY APP""")
+    st.subheader("""FOR BACKGROUND REMOVAL AND CHANGE!""")
+    col1 = None
+    col2 = None
+    final_img = None
+    with st.spinner("[UPLOAD] Image uploading"):
+        try:
+            image_file = st.file_uploader('[UPLOAD] Please upload your image:', type=["png", "jpg", "jpeg"])
+            time.sleep(1)
+        except:
+            print("[ERROR] Sorry, something went wrong!")
+            pass
+    # print(type(image_file))
+
+    if image_file is not None:
+        st.success("Load image successfully!...")
+        image = load_image(image_file)
+        # print(type(image))
+        col1, col2, col3 = st.columns(3)
+        with col1:
+            st.image(image, caption="Image before processing")
+        save_path = "tempDir/"+ image_file.name
+        image.save(save_path)
+
+
+        image_path, details = save_path, image_file
+
+        if details is not None:
+            with col2:
+                with st.spinner("[PROCESSING] Image processing"):
+                    final_img_path = main_processing(col1, col2, col3, sport_bg_path=stadium_sport_bg_path, swim_bg_path=beach_swim_bg_path,
+                                          office_bg_path=office_bg_path, img_path=image_path, name=details.name,
+                                          detection_model_path=detection_model_path,
+                                          background_model_path=background_model_path)
+                    time.sleep(1)
+
+            with col1:
+                if final_img_path is not None:
+                        final_img = load_image(final_img_path)
+                        st.image(final_img, caption="Image after processing")
+                        st.balloons()
+                        with col2:
+                            with open(final_img_path, "rb") as file:
+                                st.write('\n')
+                                st.write('\n')
+                                st.write('\n')
+                                st.write('\n')
+                                st.write('\n')
+
+                                file_name = save_path.split("/")[-1].split(".")[-2] +"_from_abc" + ".png"
+
+                                if st.download_button(
+                                    label="Download postprocessing image",
+                                    data=file,
+                                    file_name= file_name,
+                                    mime="image/png"
+                                ):
+                                    st.success('[DOWNLOAD] Download sucessfully!')
+
+
+
+if __name__ == '__main__':
+    np.random.seed(42)
+    tf.random.set_seed(42)
+
+    bg_path = ""
+    background_model_path = "weight_files/model.h5"
+    detection_model_path = "weight_files/clothes_detection_model.pt"
+
+    stadium_sport_bg_path = "backgrounds/camnou_stadium.jpg"
+    beach_swim_bg_path = "backgrounds/beach.jpg"
+    office_bg_path = "backgrounds/office-bg.jpg"
+
+    image_path = None
+
+    streamlit_app()

libs.py

@@ -0,0 +1,10 @@
+import torch
+import json
+import os
+import numpy as np
+import cv2
+import tensorflow as tf
+import streamlit as st
+import time
+from PIL import Image
+from tensorflow.keras.utils import CustomObjectScope

metrics.py

@@ -0,0 +1,25 @@
+from libs import *
+
+smooth = 1e-15
+
+
+def iou(y_true, y_pred):
+    def f(y_true, y_pred):
+        intersection = (y_true * y_pred).sum()
+        union = y_true.sum() + y_pred.sum() - intersection
+        x = (intersection + 1e-15) / (union + 1e-15)
+        x = x.astype(np.float32)
+        return x
+    return tf.numpy_function(f, [y_true, y_pred], tf.float32)
+
+
+
+def dice_coef(y_true, y_pred):
+    y_true = tf.keras.layers.Flatten()(y_true)
+    y_pred = tf.keras.layers.Flatten()(y_pred)
+    intersection = tf.reduce_sum(y_true * y_pred)
+    return (2. * intersection + smooth) / (tf.reduce_sum(y_true) + tf.reduce_sum(y_pred) + smooth)
+
+
+def dice_loss(y_true, y_pred):
+    return 1.0 - dice_coef(y_true, y_pred)

setup.sh

@@ -0,0 +1,13 @@
+mkdir -p ~/.streamlit/
+
+echo "\
+[general]\n\
+email = \"[email protected]\"\n\
+" > ~/.streamlit/credentials.toml
+
+echo "\
+[server]\n\
+headless = true\n\
+enableCORS=false\n\
+port = $PORT\n\
+" > ~/.streamlit/config.toml