Skip to content
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

a problem with the image shape of the input model #7

Closed
hutao568 opened this issue Apr 13, 2021 · 7 comments
Closed

a problem with the image shape of the input model #7

hutao568 opened this issue Apr 13, 2021 · 7 comments

Comments

@hutao568
Copy link

Hello Mingfei Chen, everything works normally when I use single GPU for training, but when I use multiple GPUs to train HOIA dataset, I encounter a problem, I would like to ask you how to solve it,there is a problem with the image shape of the input model

RuntimeError: Given groups=1, weight of size [64, 3, 7, 7], expected input[8, 1, 864, 1251] to have 3 channels, but
got 1 channels instead

@yoyomimi
Copy link
Owner

Because the evaluation code is currently working for a non-distributed environment. If you want to solve this, you can compare and refer to the train code to add the distribution-related code for evaluation.

@hutao568
Copy link
Author

I commented out the test code, but it still reports an error. In addition, I use the checkpoint and code you provided, the result file is empty, and the mAP is 0

@yoyomimi
Copy link
Owner

I commented out the test code, but it still reports an error. In addition, I use the checkpoint and code you provided, the result file is empty, and the mAP is 0

I will update the test code for distributed GPU these days. And for the empty test result, please check the log to make sure you have loaded the checkpoint successfully. If the results are still empty, please let me know. Thanks.

@hutao568
Copy link
Author

I use the test code to get the following results
image
The contents of the pred.json file are as follows
image
The md5 of the ASNet_hoia_resnet50.pth file is 5b0c2ed4f64462c0e0854e31af224ab6

@yoyomimi
Copy link
Owner

Please show the generated .log file to me and I will check that for you.

@hutao568
Copy link
Author

I checked the log file and it did not succeed in loading the model. ==> model pretrained from {resume_path}', I will try again

@hutao568
Copy link
Author

Thank you for your patient guidance, I got the same result as in the paper.
image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants