-
Notifications
You must be signed in to change notification settings - Fork 39
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Can you share the codes for generating GT IIMs? #2
Comments
Before generating IIM from dot labels, we used the NWPU dataset to train a head scale prediction model, which estimates the box size for other crowd datasets. The pre-trained model and the script file will be shared after collation. If you are in a hurry, we recommend that you can take the dilation approach, generating IIM by setting the maximum size or detecting whether the nearest areas are overlapped as the stop condition. This strategy can also generate IIM for point annotations. Thanks for your attention! |
Thanks! This is very helpful! I will try it out. |
You may process the following operation:
This operation is not scale-aware, but it is also a type of IIM. It may cause some performance reductions compared with the paper's scheme. Besides, as @taohan10200 mentioned, we will provide our size regression model ASAP. |
Thanks a lot! @gjy3035 This is quite helpful! |
|
Hi there, @gjy3035 @taohan10200 thanks a lot for your tips in generating IIMs. However, when I tried it on my own datatset(I set max width==100), the following error occurs. IndexError: index 4 is out of bounds for axis 0 with size 4, which is thrown by this line: dst_point_ = centroids[id[start + 1]]. Since I haven't quite understood your codes yet, I can't figure out what is causing this issue. (ps, it works for some images, but throws this error for others, I can't find the pattern in this error). Can you give me some tips? Thanks very much! |
Hi @gjy3035 and @taohan10200 . Thanks for this wonderful work. Are the scripts and models used to generate GT IIMs from point annotations released? Thanks. |
Yes, we now have provided the code that can directly generate IIMs with points only. You can find the function |
Hi there, thanks for this impressive work! I am interested in trying it out on my own counting dataset which uses a point level annotation. Can you share the script for getting the IIM maps from points or dot maps? Thanks very much!
Congratulations on this amazing work!
The text was updated successfully, but these errors were encountered: