search

leimao / DeepLab-V3

Google DeepLab V3 for Image Semantic Segmentation
https://leimao.github.io/project/DeepLab-V3/
MIT License
153 stars 47 forks source link

module 'tensorflow' has no attribute 'contrib' #10

Open ghost opened 3 years ago

ghost commented 3 years ago

Traceback (most recent call last): File "train.py", line 8, in from model import DeepLab File "C:\Users\DGRE\DeepLab-V3\model.py", line 6, in from feature_extractor import MobileNet, Resnet, Vgg16 File "C:\Users\DGRE\DeepLab-V3\feature_extractor.py", line 2, in from nets import resnet_v2 File "C:\Users\DGRE\DeepLab-V3\nets\resnet_v2.py", line 55, in from nets import resnet_utils File "C:\Users\DGRE\DeepLab-V3\nets\resnet_utils.py", line 43, in slim = tf.contrib.slim AttributeError: module 'tensorflow' has no attribute 'contrib'

Vargha-Kh commented 3 years ago

Use Tensorflow-gpu=1.15.0 requirement.txt will install tensorflow=2.4.0