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

Evaluate with SUNRGBD Dataset with the weight file which is trained on Scenenet Dataset #36

Open
suesuekkim opened this issue Jun 6, 2022 · 1 comment

Comments

@suesuekkim
Copy link

suesuekkim commented Jun 6, 2022

I have some questions about your model ESANet.

I trained your ESANet model on the scenenetrgbd dataset. And I want to evaluate the weight file (which calculated through scenenetrgbd dataset) in the SUNRGBD dataset. But the scenenet dataset has only 13 classes, while SUNRGBD dataset has 37 classes. Due to the number of classes are mismatched, the error occured. Is it possible to evaluate the scenenet pretrained weight file to SUNRGBD dataset?

Thank you for reading my issue.

@suesuekkim suesuekkim changed the title Can we implement ESANet with custom datasets? Evaluate with SUNRGBD Dataset with the weight file which is trained on Scenenet Dataset Jun 8, 2022
@danielS91
Copy link
Member

SceneNetRGBD is a synthetic dataset quite far away from reality in terms of physics and scene layout. IMHO it is only useful for pretraining in order to initialize the weights in better way than random weights would do. It does not really make sense to evaluate on SUNRGBD after training on SceneNetRGBD solely.

However, SceneNetRGBD follows the NYUv2 class spectrum with 13 classes, some of the classes match classes in the 40 (= 37 + 3 otherclasses, while the latter ones are mapped to void in SUNRGBD) class spectrum of NYUv2. You can create a simple mapping based on the class names. Moreover, you can have a look on the forward mapping from 40 to 13 classes during dataset preparation HERE.
Nevertheless, the remaining classes are missing and cannot be predicted in your setting.

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