search

whai362 / PSENet

Official Pytorch implementations of PSENet.
Apache License 2.0
1.17k stars 344 forks source link

what is the difference between kernal_num and num_classes? #158

Open Zhang-O opened 4 years ago

Zhang-O commented 4 years ago

https://github.com/whai362/PSENet/blob/d5926688a531242acd5bd5b771c94d1998bac4dd/test_ctw1500.py#L205
I am confused about the help info here.

kernal_num should be same as num_class in "model = models.resnet50(pretrained=True, num_classes=kernel_num, scale=scale)"?