When running your code I got a error

[usalexsantos]

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?

[susheelsk]

Looks like a virtualenv issue. Can you try reinstalling tf?

[dirkk0]

pip3 install tensorflow==1.14 fixes this.

[NMNaufaldo]

@dirkk0 In my case, I've to downgrade Python 3.8 to 3.7 to be able to install tensorflow==1.14

[dome272]

For those that might have the same issue and dont want to downgrade anything I found this solution:
tensorflow/models#7703 (comment)

[SaeedArisha]

instead of downgrading the Tf version trace the error, go to the line causing it, and then write tf.compat.v1.GraphDef

[adills]

Needed to make the following 3 changes in the seg.py file to be compatible with Py 3.11.6, tf 2.16.1, and pillow 10.2.0:

1. line 25, tf.GraphDef.FromString tf.compat.v1.GraphDef.FromString
2. line 33, tf.Session to tf.compat.v1.Session
3. line 50, Image.ANTIALIAS to Image.LANCZOS

See updated file attached. Change the .txt back to .py after downloading.
seg.txt