fasterrcnn_mobilenet_v3_large_fpn num_classes expected value 91 but got 20 instead

[ardeal]

Hi,

I experienced the following error when I run demo.py:

Exception has occurred: ValueError
The parameter 'num_classes' expected value 91 but got 20 instead.
  File "C:\doc\code_python\AA\hukenovs_hagrid\detector\models\fasterrcnn.py", line 10, in __init__
    torchvision_model = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn(
  File "C:\doc\code_python\AA\hukenovs_hagrid\detector\utils.py", line 138, in build_model
    "FasterRCNN_mobilenet_large": FasterRCNN_Mobilenet_large(pretrained=pretrained, num_classes=num_classes),
  File "C:\doc\code_python\AA\hukenovs_hagrid\demo.py", line 149, in <module>
    model = build_model(
ValueError: The parameter 'num_classes' expected value 91 but got 20 instead