search

VainF / DeepLabV3Plus-Pytorch

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes
MIT License
2.01k stars 448 forks source link

Issue with NNPack #95

Open jacobsayono opened 2 years ago

jacobsayono commented 2 years ago

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!

VainF commented 2 years ago

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

jacobsayono commented 1 year ago

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?

jacobsayono commented 1 year ago

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?