search

AlexMa011 / pytorch-polygon-rnn

Pytorch implementation of Polygon-RNN(http://www.cs.toronto.edu/polyrnn/poly_cvpr17/)
GNU General Public License v3.0
113 stars 43 forks source link

Training is throwing error. #11

Closed vineetjai closed 5 years ago

vineetjai commented 5 years ago

I am getting error while doing training. Please help me:

$ python train.py --gpu_id 0 --batch_size 8  --lr 0.0001
Traceback (most recent call last):
  File "train.py", line 12, in <module>
    from test import test
  File "/home/ai/Documents/vineet/pytorch-polygon-rnn/test.py", line 14, in <module>
    from utils.utils import img2tensor
ImportError: No module named 'utils.utils'

I think there has to be file(utils.py) in utils folder which consist of img2tensor, iou and getbboxfromkps. So please include function in utils folder or change test.py such that

from utils.utils import img2tensor
from utils.utils import iou, getbboxfromkps

can be replaced by other function. Or am I need to install packages other than requirements package? please correct me if I am wrong.

AlexMa011 commented 5 years ago

Sorry about that. I will try to update to fix the error as. It was my mistake to forget to add the utils file.

AlexMa011 commented 5 years ago

I already update the project, please try it again.

vineetjai commented 5 years ago

Also can you share your pretrained model?

vineetjai commented 5 years ago

As python train.py --gpu_id 0 --batch_size 8 --lr 0.0001 is throwing error FileNotFoundError: [Errno 2] No such file or directory: './checkpoint/train1_20.pth'. So please upload your pretrained model also.