-
Notifications
You must be signed in to change notification settings - Fork 386
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
how to change the input picture's dimension #20
Comments
@msha096 Can I see your full log? |
Yes here it is. If I change to [1024,2048], it will be out of my cuda's capability.So I changed to 512,1024. (CenterNet) ➜ CenterNet-master python train.py CenterNet-104 Traceback (most recent call last): |
@msha096 Add these codes above the
|
@Duankaiwen why not update the code on repos |
Hi, I am trying to use my own dataset and the image is 1024*2048. I changed the 'input size' of 'db', in config/CenterNet104.json. But I got the following errors:
Traceback (most recent call last):
File "train.py", line 207, in
train(training_dbs, validation_db, args.start_iter)
File "train.py", line 139, in train
training_loss, focal_loss, pull_loss, push_loss, regr_loss = nnet.train(**training)
File "/home/mingzhi/Downloads/CenterNet-master/nnet/py_factory.py", line 82, in train
loss_kp = self.network(xs, ys)
File "/home/mingzhi/anaconda3/envs/CenterNet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/mingzhi/Downloads/CenterNet-master/models/py_utils/data_parallel.py", line 68, in forward
return self.module(*inputs[0], **kwargs[0])
File "/home/mingzhi/anaconda3/envs/CenterNet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/mingzhi/Downloads/CenterNet-master/nnet/py_factory.py", line 21, in forward
loss_kp = self.loss(preds, ys, **kwargs)
File "/home/mingzhi/anaconda3/envs/CenterNet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/mingzhi/Downloads/CenterNet-master/models/py_utils/kp.py", line 339, in forward
pull, push = self.ae_loss(tl_tag, br_tag, gt_mask)
File "/home/mingzhi/Downloads/CenterNet-master/models/py_utils/kp_utils.py", line 224, in _ae_loss
dist = tag_mean.unsqueeze(1) - tag_mean.unsqueeze(2)
RuntimeError: Dimension out of range (expected to be in range of [-2, 1], but got 2)
The text was updated successfully, but these errors were encountered: