Discrepancy in number of classes in the configuration file

DevashishPrasad / CascadeTabNet

This repository contains the code and implementation details of the CascadeTabNet paper "CascadeTabNet: An approach for end to end table detection and structure recognition from image-based documents"

MIT License

1.46k stars 422 forks source link

Discrepancy in number of classes in the configuration file #96

Open faizan1041 opened 3 years ago

faizan1041 commented 3 years ago

I was trying to train the model on the configuration file provided here Config/cascade_mask_rcnn_hrnetv2p_w32_20e.py. I'm failing to understand the number of classes, considering bordered, borderless and cell, it should be 3+1 classes instead of 81 which is here. Can someone provide any explanation on this?

marcelodiaz558 commented 3 years ago

Hello, I have the same doubt, did you figured it out?

faizan1041 commented 3 years ago

Yes, that file is a sample from somewhere else, the real number of classes should be (num_of_classes + 1)

DB11051998 commented 1 year ago

https://github.com/open-mmlab/mmdetection/issues/6340

I guess this might clear your doubts!!