image-segmentation-keras

 image-segmentation-keras 是GitHub上开源的图像语义分割模型框架。

本环境搭建基于 英伟达 T4 的环境使用

tensorflow-gpu 2.4.1 keras.2.3.4  CUDA 11.2

https://github.com/divamgupta/image-segmentation-keras


实验中发现,tensorflow是比较新的,但是 keras比较旧。keras里边用的可能是旧版本的tensoflow
,如果要用新版的tensorflow-gpu 。
可将程序中的 impor keras 换成 import tensorflow.keras
例如

from keras.models import load_model 改成
from tensorflow. keras.models import load_model
这样可以避免 tf2 等使用tensorflow 低版本的错误。

会发生在使用高版本的tensorflow-gpu 2.4.1

sitemap