EdVince / CLIP-ImageSearch-NCNN

CLIP⚡NCNN⚡基于自然语言的图片搜索(Image Search)⚡以字搜图⚡x86⚡Android
220 stars 20 forks source link

Trouble with larger image datasets in project testing #6

Open RealTomNook opened 1 year ago

RealTomNook commented 1 year ago

I was studying your project, and was able to run your sample compiled code against the sample images provided, and that worked fine. Then I created my own sample dataset of about 10 images and that also worked. But when I tried it with 1000 pictures, it didn't work. I got a lot of weird results that didn't make sense. I'm not sure how to fix it so that it can work with more pictures. From debugging the code, it seems when running this number of images, I get a lot of NaN values for softmax. Was wondering if there's anything that can be tweaked to improve this?