diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..4265746 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +venv/**/** diff --git a/images/input.jpg b/images/input.jpg new file mode 100644 index 0000000..c27c518 Binary files /dev/null and b/images/input.jpg differ diff --git a/images/processed.jpg b/images/processed.jpg new file mode 100644 index 0000000..2bcc00d Binary files /dev/null and b/images/processed.jpg differ diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..b3fc015 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,37 @@ +attrs==23.2.0 +certifi==2024.6.2 +charset-normalizer==3.3.2 +coloredlogs==15.0.1 +flatbuffers==24.3.25 +greenlet==3.0.3 +humanfriendly==10.0 +idna==3.7 +imageio==2.34.2 +jsonschema==4.22.0 +jsonschema-specifications==2023.12.1 +lazy_loader==0.4 +llvmlite==0.43.0 +mpmath==1.3.0 +msgpack==1.0.8 +networkx==3.3 +numba==0.60.0 +numpy==1.26.4 +onnxruntime==1.18.1 +opencv-python-headless==4.10.0.84 +packaging==24.1 +pillow==10.4.0 +platformdirs==4.2.2 +pooch==1.8.2 +protobuf==5.27.2 +PyMatting==1.1.12 +pynvim==0.5.0 +referencing==0.35.1 +rembg==2.0.57 +requests==2.32.3 +rpds-py==0.18.1 +scikit-image==0.24.0 +scipy==1.14.0 +sympy==1.12.1 +tifffile==2024.7.2 +tqdm==4.66.4 +urllib3==2.2.2 diff --git a/scripts/background.py b/scripts/background.py new file mode 100644 index 0000000..146da8d --- /dev/null +++ b/scripts/background.py @@ -0,0 +1,10 @@ +from rembg import remove + +in_file = '../images/input.jpg' +out_file = '../images/processed.jpg' + +with open(in_file, 'rb') as i: + with open(out_file, 'wb') as o: + in_b = i.read() + out = remove(in_b) + o.write(out)