inria-aerial-image-labeling icon indicating copy to clipboard operation
inria-aerial-image-labeling copied to clipboard

Published 20 hours ago • verified publisher icon a-milosavljevic

about requirements problems

Open Moyao6 opened this issue 4 years ago • 0 comments

Thanks for sharing your code. When I tried to use the given requirements, there were some problems occurs. I managed to train with these version: numpy: 1.18.5 tensorflow-gpu: 2.3.0 keras: 2.4.3 segmentation-models: 1.0.1 CUDA Version 10.1.243

you can use this images to get all these environments: docker pull smitharauco/tensorflow2.3gpu

After using these environments, you should insert one line of code in model.py from keras.utils import generic_utils before: import segmentation_models as sm

Moyao6 avatar Feb 02 '21 08:02 Moyao6