AI Remove image background(use rembg)

colinusa · **Joined:** Dec 13, 2022 **Posts:** 4

Thank you JamesH

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

Hi, I am able to remove the background using rembg CLI. But not through GIMP 2.10.25 AppImage using this plugin in Ubuntu 22.04.

The plugin works but nothing changes in the canvas. And, when I removed the part where it deletes the tmp folder entry I am able to see the input picture as-is in the tmp folder.

My guess is the rembg is not applied at all due to some reason.

Adding the plugin code I have. I modified it a bit to add my rembg entry and use the AppImage from /usr/bin

Code: Select all

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# James Huang <[email protected]>
# https://elastic192.blogspot.com/

from gimpfu import *
import os, sys, string, tempfile
import platform

def python_fu_RemoveBG(image, drawable, asMask, AlphaMatting, aeValue):
   
   osName = platform.system()
   exportSep = str(os.sep)
   tdir = tempfile.gettempdir()
   #tdir = "./"
   print(tdir)
   jpgFile = "%s%sTemp-gimp-0000.jpg" % (tdir, exportSep)
   pngFile = "%s%sTemp-gimp-0000.png" % (tdir, exportSep)
   x1 = 0
   y1 = 0
   option = ""

   image.undo_group_start()
   curLayer = pdb.gimp_image_get_active_layer(image)

   if pdb.gimp_selection_is_empty(image):
      pdb.file_jpeg_save(image, drawable, jpgFile, jpgFile, 0.95, 0, 1, 0, "", 0, 1, 0, 0)
   else:
      pdb.gimp_edit_copy(drawable)
      non_empty, x1, y1, x2, y2 = pdb.gimp_selection_bounds(image)
      tmpImage = gimp.Image(x2-x1, y2-y1, 0)
      tmpDrawable = gimp.Layer(tmpImage, "Temp", tmpImage.width, tmpImage.height, RGB_IMAGE, 100, NORMAL_MODE)
      pdb.gimp_image_add_layer(tmpImage, tmpDrawable, 0)
      pat = pdb.gimp_context_get_pattern()
      pdb.gimp_context_set_pattern("Leopard")
      pdb.gimp_drawable_fill(tmpDrawable, 4)
      pdb.gimp_context_set_pattern(pat)
      pdb.gimp_floating_sel_anchor(pdb.gimp_edit_paste(tmpDrawable,TRUE))
      pdb.file_jpeg_save(tmpImage, tmpDrawable, jpgFile, jpgFile, 0.95, 0, 1, 0, "", 0, 1, 0, 0)
      pdb.gimp_image_delete(tmpImage)
   
   aiExe = "/home/demo/.local/bin/rembg"
   if AlphaMatting:
      option = "-a -ae %d" % (aeValue)
   cmd = "gimp25.AppImage --host %s i %s %s %s" % (aiExe, option, jpgFile, pngFile)
   os.system(cmd)

   file_exists = os.path.exists(pngFile)
   if file_exists:
      newlayer = pdb.gimp_file_load_layer(image, pngFile)
         image.add_layer(newlayer, -1)
         pdb.gimp_layer_set_offsets(newlayer, x1, y1)
         if asMask:
            pdb.gimp_image_select_item(image, CHANNEL_OP_REPLACE, newlayer)
            image.remove_layer(newlayer)
         copyLayer = pdb.gimp_layer_copy(curLayer, TRUE)
         image.add_layer(copyLayer, -1)
         mask=copyLayer.create_mask(ADD_SELECTION_MASK)
         copyLayer.add_mask(mask)
         pdb.gimp_selection_none(image)

      image.undo_group_end()
      gimp.displays_flush()

   

register(
    "python_fu_RemoveBG",
    "AI移除影像背景, AI Remove image background",
    "AI移除影像背景, AI Remove image background",
    "JamesH, <[email protected]>",
    "JamesH, https://elastic192.blogspot.com",
    "2022/6/4",
    "<Image>/Python-Fu/AI移除影像背景(AI Remove background) ...",
    "RGB*",
    [
       (PF_TOGGLE, "asMask", ("當作圖層遮罩(as Mask)"), True),
       (PF_TOGGLE, "AlphaMatting", ("alpha matting"), False),
       (PF_SPINNER,"aeValue", ("ALPHA_MATTING_ERODE_SIZE"), 15, (15,100,1))
    ],
    [],
    python_fu_RemoveBG,
    domain=("gimp20-python", gimp.locale_directory))

main()

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

Hi, I am able to remove the background using rembg CLI. But not through GIMP 2.10.25 AppImage using this plugin in Ubuntu 22.04.

The plugin works but nothing changes in the canvas. And, when I removed the part where it deletes the tmp folder entry I am able to see the input picture as-is in the tmp folder.

My guess is the rembg is not applied at all due to some reason.

Adding the plugin code I have. I modified it a bit to add my rembg entry and use the AppImage from /usr/bin

Any help will be great.

JamesH · **Posted:** Tue Jan 17, 2023 3:35 am **(#34)**

meetdilip wrote:

Hi, I am able to remove the background using rembg CLI. But not through GIMP 2.10.25 AppImage using this plugin in Ubuntu 22.04.

The plugin works but nothing changes in the canvas. And, when I removed the part where it deletes the tmp folder entry I am able to see the input picture as-is in the tmp folder.

My guess is the rembg is not applied at all due to some reason.

for linux AppImage gimp
RemoveBG-AppImage.zip
find
aiExe = "/home/jamesh/.local/bin/rembg"
Replace with your own rembg path

Test successful on ubuntu 22.04 GIMP_AppImage-git-2.10.25-20210610-withplugins-x86_64.AppImage

racer-x · **Joined:** Apr 15, 2017 **Posts:** 1837

Finally a RemBG plugin for Gimp that actually works. It works great in PPA installed Gimp, the other ones wouldn't work. I like the option of using Mask, much more useful that way.

Thanks JamesH for posting the last one.

uglybeauty · **Joined:** Jan 22, 2023 **Posts:** 1

This is pretty cool, thanks! I'll have to check out backgroundremover too.

In case anyone is still having trouble installing this on Windows, I wrote up a small guide as it took me a little while to figure out.

SHOW SPOILER

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

Hi, I added the .py file in plugin folder for both AppImage and Flatpak (and run as instructed). But I do not see any option to use rembg inside GIMP. Where can I find it? Thanks.

Edit 1:

Tried CLI, it is working. I am on Ubuntu 22.04

Jappaloe63 · **Posted:** Fri May 19, 2023 7:40 am **(#38)**

I was so happy with remove background, but somehow he doesn't do it anymore. Can it still be repaired?

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

Hi, I tried rembg install on a new Ubuntu. Sadly, it is giving the following error

Code: Select all

$ rembg --help
Traceback (most recent call last):
  File "/home/myPC/.local/bin/rembg", line 5, in <module>
    from rembg.cli import main
  File "/home/myPC/.local/lib/python3.10/site-packages/rembg/__init__.py", line 5, in <module>
    from .bg import remove
  File "/home/myPC/.local/lib/python3.10/site-packages/rembg/bg.py", line 6, in <module>
    from cv2 import (
ImportError: cannot import name 'BORDER_DEFAULT' from 'cv2' (unknown location)

Not sure what is wrong.

Started a bug report as well

https://github.com/danielgatis/rembg/issues/515

Any help will be great. Thanks

Tas_mania · **Joined:** Nov 04, 2015 **Posts:** 1459

Looks like you need opencv-python. You may have to trick it into believing it's installed

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

Thanks Tas_mania

I got stuck here though

Quote:

$ mkdir ~/.conda/envs/orion/lib/python3.8/opencv_python-3.4.11.dist-info
mkdir: cannot create directory ‘/home/myPC/.conda/envs/orion/lib/python3.8/opencv_python-3.4.11.dist-info’: No such file or directory

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

I installed Open CV using

Code: Select all

sudo apt install python3-opencv -y

but still getting an error

Quote:

rembg i /home/myPC/12.png /home/myPC/haa.png
Traceback (most recent call last):
File "/home/myPC/.local/bin/rembg", line 8, in <module>
sys.exit(main())
File "/home/myPC/.local/lib/python3.10/site-packages/rembg/cli.py", line 33, in main
_main() # type: ignore
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 829, in __call__
return self.main(*args, **kwargs)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 782, in main
rv = self.invoke(ctx)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 1257, in invoke
sub_ctx = cmd.make_context(cmd_name, args, parent=ctx)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 700, in make_context
self.parse_args(ctx, args)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 1048, in parse_args
value, args = param.handle_parse_result(ctx, opts, args)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 1623, in handle_parse_result
value = self.full_process_value(ctx, value)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 1965, in full_process_value
return Parameter.full_process_value(self, ctx, value)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 1592, in full_process_value
value = self.get_default(ctx)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 1917, in get_default
return Parameter.get_default(self, ctx)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 1534, in get_default
return self.type_cast_value(ctx, rv)
File "/home/myPC/.local/lib/python3.10/site-packages/click/core.py", line 1561, in type_cast_value
return self.type(value or (), self, ctx)
File "/home/myPC/.local/lib/python3.10/site-packages/click/types.py", line 46, in __call__
return self.convert(value, param, ctx)
File "/home/myPC/.local/lib/python3.10/site-packages/click/types.py", line 681, in convert
raise TypeError(
TypeError: It would appear that nargs is set to conflict with the composite type arity.

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

Solved using this trick. Commented out a few lines in i_command.py

https://github.com/danielgatis/rembg/is ... 1621412245

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

A different PC, installing on Windows 10. Added rembg, the GPU version + CLI, added the correct path and placed the plugin in plugins folder. Nothing under PythonFU. Restarted the computer, still nothing. Do we need to use some special command to start GIMP in Windows 10 to get the bg remove options? Thanks

Using 2.10.24

meetdilip · **Joined:** Nov 19, 2019 **Posts:** 209

Found a solution. It is not working when added under program files. So added under AppData path. Now fine

contrast_ · **Joined:** Oct 31, 2020 **Posts:** 2013

This plugin will soon have a Linux only GEGL counterpart that calls REMBG or any bash string from the users system in GEGL.

viewtopic.php?f=9&t=20652

On my machine for it to work I will have to
1.open the GEGL plugin.
2. type in a given import/export file name
3.insert the string

Code: Select all

source ~/Applications/mc/etc/profile.d/conda.sh && conda activate base && rembg -p

Then REMBG will be inside GEGL due to it being a bash string. Though I must note Gimp will be frozen until REMBG loads, Stay tuned!

nelo · **Posted:** Mon Nov 27, 2023 1:36 pm **(#47)**

@contrast_ that sounds really cool. :coolthup

contrast_ · **Joined:** Oct 31, 2020 **Posts:** 2013

outdated post

contrast_ · **Joined:** Oct 31, 2020 **Posts:** 2013

REMBG now runs in GEGL and you guys can see it here.

viewtopic.php?f=9&t=20652&start=10

contrast_ · **Joined:** Oct 31, 2020 **Posts:** 2013

JamesH was the one who wrote the flatpak-spawn stuff but I edited it and got it working on the latest release of REMBG from late 2023. Please replace contrast with your username

Code: Select all

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# James Huang <[email protected]>
# https://elastic192.blogspot.com/

from gimpfu import *
import os, sys, string, tempfile
import platform

def python_fu_RemoveBG(image, drawable, asMask, AlphaMatting, aeValue):
   removeTmpFile = True
   osName = platform.system()
   exportSep = str(os.sep)
   tdir = "/home/contrast/rembggimp"
   #tdir = "./"
   print(tdir)
   jpgFile = "%s%ssubjectR.jpg" % (tdir, exportSep)
   pngFile = "%s%ssubjectR.png" % (tdir, exportSep)
   x1 = 0
   y1 = 0
   option = ""

   image.undo_group_start()
   curLayer = pdb.gimp_image_get_active_layer(image)

   if pdb.gimp_selection_is_empty(image):
      pdb.file_jpeg_save(image, drawable, jpgFile, jpgFile, 0.95, 0, 1, 0, "", 0, 1, 0, 0)
   else:
      pdb.gimp_edit_copy(drawable)
      non_empty, x1, y1, x2, y2 = pdb.gimp_selection_bounds(image)
      tmpImage = gimp.Image(x2-x1, y2-y1, 0)
      tmpDrawable = gimp.Layer(tmpImage, "Temp", tmpImage.width, tmpImage.height, RGB_IMAGE, 100, NORMAL_MODE)
      pdb.gimp_image_add_layer(tmpImage, tmpDrawable, 0)
      pat = pdb.gimp_context_get_pattern()
      pdb.gimp_context_set_pattern("Leopard")
      pdb.gimp_drawable_fill(tmpDrawable, 4)
      pdb.gimp_context_set_pattern(pat)
      pdb.gimp_floating_sel_anchor(pdb.gimp_edit_paste(tmpDrawable,TRUE))
      pdb.file_jpeg_save(tmpImage, tmpDrawable, jpgFile, jpgFile, 0.95, 0, 1, 0, "", 0, 1, 0, 0)
      pdb.gimp_image_delete(tmpImage)
   
   aiExe = "/home/contrast/.local/bin/rembg"
   if AlphaMatting:
      option = "-a -ae %d" % (aeValue)
   cmd = "flatpak-spawn --host %s i %s %s" % (aiExe, jpgFile, pngFile)
   os.system(cmd)

   file_exists = os.path.exists(pngFile)
   if file_exists:
      newlayer = pdb.gimp_file_load_layer(image, pngFile)
         image.add_layer(newlayer, -1)
         pdb.gimp_layer_set_offsets(newlayer, x1, y1)
         if asMask:
            pdb.gimp_image_select_item(image, CHANNEL_OP_REPLACE, newlayer)
            image.remove_layer(newlayer)
         copyLayer = pdb.gimp_layer_copy(curLayer, TRUE)
         image.add_layer(copyLayer, -1)
         mask=copyLayer.create_mask(ADD_SELECTION_MASK)
         copyLayer.add_mask(mask)
         pdb.gimp_selection_none(image)

      image.undo_group_end()
      gimp.displays_flush()

   if removeTmpFile:
      if osName == "Windows":
         del_command = "del \"%s%ssubjectR.*\"" % (tdir, exportSep)
      else:
         del_command = "sleep 0.6 && rm %s%ssubjectR.*" % (tdir, exportSep)
      os.system(del_command)

register(
    "python_fu_RemoveBG",
    "AI移除影像背景, AI Remove image background",
    "AI移除影像背景, AI Remove image background",
    "JamesH, <[email protected]>",
    "JamesH, https://elastic192.blogspot.com",
    "2022/6/4",
    "<Image>/REMBG/AI移除影像背景(AI Remove background) ...",
    "RGB*",
    [
       (PF_TOGGLE, "asMask", ("當作圖層遮罩(as Mask)"), False),
       (PF_TOGGLE, "AlphaMatting", ("alpha matting"), False),
       (PF_SPINNER,"aeValue", ("ALPHA_MATTING_ERODE_SIZE"), 10, (10,100,1))
    ],
    [],
    python_fu_RemoveBG,
    domain=("gimp20-python", gimp.locale_directory))

main()