syurchen / underlords

Dota Underlords helper WIP. Live demo:
MIT License
2 stars 0 forks source link

Try more efficient way to detect hero thumbnails? #2

Open Umiiii opened 5 years ago

Umiiii commented 5 years ago

I've noticed that you are using OpenCV to detect every hero, but it seems slow and buggy. Have you ever try another way to detect it? Such as machine learning or hash distance calculation.

syurchen commented 5 years ago

Yeah, detecting is not good. I will probably try something different in near future. Can you recommend examples I can follow or literature I can read on topic?

Umiiii commented 5 years ago

https://developer.apple.com/documentation/createml/creating_an_image_classifier_model Here is a simple example, and I tried with a little small dataset, so the model overfits, but got better performance. 2 (The failed one) 1 (The successful one)

The dataset is transformed by GaussianBlur first, here is the reason: WX20190913-223838

Umiiii commented 5 years ago

Please notice that there is no need to use the Apple ML framework. You may just train the classifier using another deep learning framework, I used because it is out-of-the-box.

syurchen commented 5 years ago

Thanks, I will try pytorch in separate branch. Let's see, if I will be able to come up with anything good