VainF / DeepLabV3Plus-Pytorch

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Issue with NNPack

jacobsayono opened this issue · comments

Hi,

I'm using a VM VirtualBox with Ubuntu 20.04 LTS, I get this error when I try to run the single image prediction.

[W NNPACK.cpp:53] Could not initialize NNPACK! Reason: Unsupported hardware.

Thanks!

Hi @jacobsayono, it looks like a VirtualMachine-related issue. Maybe NNPack used by Pytorch is not supported by your VM?

Thanks for the reply! @VainF
I was using VirtualBox (which gave this error), I setup a new virtual machine using VMWare this week.
Looks like it is a warning from VirtualBox; VMWare does not give that warning.

Some things I learned is that VirtualBox has a 128 Mb video memory cap while VMWare has 2 Gb... do you know if it is something related to this?

sidewalk
sidewalk

Looks like it's working now; I fixed some paths for both the .pth file and input images. Here's the output segmentation of a sidewalk!
I am wondering if the bleeds are normal, and how I can go about improving the segmentation? Do you think the model is currently overfitting? Or should I try to feed in more CityScapes training data?