K out of range - Githubissues

OlafenwaMoses / ImageAIDocumentation

Official English Documentation of ImageAI

MIT License

23 stars 30 forks source link

K out of range #19

Open BillJohns opened 8 months ago

BillJohns commented 8 months ago

File "/Users/billjohnsmaci9/anaconda3/lib/python3.11/site-packages/imageai/Classification/Custom/init.py", line 542, in classifyImage topN_prob, topN_catid = torch.topk(probabilities, result_count) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: selected index k out of range

How can this be fixed?