deeplab_v3
Deeplabv3 +的Keras实现
Deeplabv3 +的Keras实现
DeepLab是用于语义图像分割的最先进的深度学习模型。
模型基于原始的TF冻结图。 可以将预训练的权重加载到此模型中。 权重直接从原始TF检查点导入。
如何获得标签
模型将返回形状的张量(batch_size,height,width,classes)。 要获取标签,您需要将argmax应用于退出层的logits。 在image1.jpg上进行预测的示例:
from matplotlib import pyplot as plt
import cv2 # used for resize. if you dont have it, use anything else
import numpy as np
from model import Deeplabv3
deeplab_mod
2021-07-05 19:32:21
9KB
Python
1