susheelsk / image-background-removal Goto Github PK

hi
thanks for sharing your code, i was searching for a complete solution like yours for a while.
the outputs are pretty good , but i was looking for a model to give me high definition output.

could you please guide me where i can find the models ?
thanks again

python seg.py single1.jpg sample5.png 1

Output
Killed

thanks for sharing sir,
but i want to implement this functionality using mask rcnn model,
please help me,

$python3 seg.py sample.jpg sample.png

Traceback (most recent call last):
File "seg.py", line 89, in
MODEL = DeepLabModel(modelType)
File "seg.py", line 25, in init
graph_def = tf.GraphDef.FromString(open(tarball_path + "/frozen_inference_graph.pb", "rb").read())
AttributeError: module 'tensorflow' has no attribute 'GraphDef'

I am getting this error when execute the given command. Can you please help me on this?

How I can use your code in Android app using JAVA platform? Please help me if possible.

Hi @susheelsk, I got an error running your code.

C:\image-background>python seg.py /input/0001.jpg /output/0001.png 1
Traceback (most recent call last):
File "seg.py", line 89, in
MODEL = DeepLabModel(modelType)
File "seg.py", line 25, in init
graph_def = tf.GraphDef.FromString(open(tarball_path + "/frozen_inference_graph.pb", "rb").read())
AttributeError: module 'tensorflow' has no attribute 'GraphDef'

Can you help?

I tried to change the input size from INPUT_SIZE = 513. but i got an error

Is there a way to make the output PNG the same size than the input picture?

On line no 79

platform - googlecollab

Hello,
I executed the program according to README guide, but didn't get expected result, is there anything missing?