yoyomimi / AS-Net

Code for one-stage adaptive set-based HOI detector AS-Net.
MIT License
48 stars 1 forks source link

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

Closed hutao568 closed 3 years ago

hutao568 commented 3 years ago

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 commented 3 years ago

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 commented 3 years ago

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 commented 3 years ago

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 commented 3 years ago

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 commented 3 years ago

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

hutao568 commented 3 years ago

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

hutao568 commented 3 years ago

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