关于训练结果 about deeplabv3plus-pytorch HOT 7 OPEN

I meet the same bug. If there has any update, please tell me ASAP. Thanks a lot!

from deeplabv3plus-pytorch.

I meet the same bug. If there has any update, please tell me ASAP. Thanks a lot!

from network import *
from torchvision.transforms.transforms import *
from datasets.cityscapes import *

val_transform = Compose([
ToTensor(),
Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225]),
])
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model_path = 'checkpoints/latest_deeplabv3_resnet50_cityscapes_os16.pth'
model = deeplabv3_resnet50(num_classes=19,output_stride=16)

model.load_state_dict(torch.load(model_path)['model_state'])
model.to(device)
model.eval()

img_path = 'results/4_image.png'
image = Image.open(img_path).convert('RGB')
input = cv2.cvtColor(np.asarray(image),cv2.COLOR_RGB2BGR)
if name == 'main':
import torch

cv2.namedWindow('',0)

model_dict = torch.load(model_path)
test_input = val_transform(image).unsqueeze(dim=0)
test_input = test_input.to(device)
print('输入图像：',test_input.size())
output =model(test_input).cpu().detach().clone()
print('输出:',output.size())

preds = output.max(dim=1)[1].numpy()
print(preds)
pred = preds[0]
pred = Cityscapes.decode_target(pred).astype(np.uint8)
cv2.imshow('',pred)

cv2.waitKey(0)

from deeplabv3plus-pytorch.

问题已解决 haha!

from deeplabv3plus-pytorch.

from deeplabv3plus-pytorch.

楼上是怎么解决的？

from deeplabv3plus-pytorch.

@KOOKOKOK

from deeplabv3plus-pytorch.

问题已解决 haha!

大哥 请问作者的代码 更新解决了这个问题没？若解决 请告知 万分感谢 万分感谢！！！

from deeplabv3plus-pytorch.