search

RangiLyu / nanodet

NanoDet-Plusāš”Super fast and lightweight anchor-free object detection model. šŸ”„Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphonešŸ”„
Apache License 2.0
5.78k stars 1.04k forks source link

Google Colab train.py fails due to "No module named 'nanodet'" #439

Closed Vivinternational closed 2 years ago

Vivinternational commented 2 years ago

Hi there,

I am trying to set up a training on Google Colab, but even though after "!python "/content/drive/MyDrive/nanodet/setup.py" develop" works fine and "!pip list -v | grep nanodet" shows me nanodet exists and trying to install again it says "Requirements already satisfied", I always get the info "No module named 'nanodet'" found. Also trying to import nanodet does not work. Restart of kernel also did not work.

Could anybody help. please?!

Setting it up on my local machine works fine, but I do not have GPU for training...

Thanks a lot!

Vivinternational commented 2 years ago

If I clone it directly into the Google Colab Jupyter Notebook with "!git clone https://github.com/RangiLyu/nanodet.git" and put in "!python setup.py develop" it works and I can train it with "/content/nanodet/tools/train.py" "/content/drive/MyDrive/nanodet-plus-m_320.yml". Alas it is not lying in the folder I wanted it to be, but at least it is working this way! :-)