detecto icon indicating copy to clipboard operation
detecto copied to clipboard

Annotations is maximum 100?

Open jerone-1997 opened this issue 4 years ago • 1 comments

Describe the bug While using Detecto library, the predictions is it always giving out a maximum of 100 annotations that is detected when training a custom object?

Code and Data Share the code that caused the error as well as an example image/label from your dataset.

Environment:

  • OS:
  • Python version:
  • Detecto version:
  • torch version:
  • torchvision version:

Additional context Add any other context about the problem here.

jerone-1997 avatar Jun 10 '21 03:06 jerone-1997

It looks like this might be the case by default with PyTorch's models. I'd recommend taking a look at this thread and seeing if you can manually override that option with your model (you can access the underlying torchvision model using the core.Model.get_internal_model method)

alankbi avatar Jun 13 '21 02:06 alankbi