thuyngch / Human-Segmentation-PyTorch

Human segmentation models, training/inference code, and trained weights, implemented in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ModuleNotFoundError: No module named 'timm.models.gen_efficientnet'

ivan-v-kush opened this issue · comments

get such error,
installed timm using pip3 install timm

run python3 inference_webcam.py --checkpoint UNet_MobileNetV2.pth

commented

get such error,
installed timm using pip3 install timm

run python3 inference_webcam.py --checkpoint UNet_MobileNetV2.pth

may i ask you how to solve this problem ?

sorry, but i've forgot already =(

commented

I degrade the version of timm solved this issue.

What version of timm is required?

pip3 install timm==0.1.10 worked for me on ubuntu 18.04

It is also enough to do

git submodule sync
git submodule update --init --recursive
pip install -e models/pytorch-image-models

as the required version of timm is incorporated in this repo.