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