ammirato / target_driven_instance_detection

MIT License
33 stars 10 forks source link

Add attention instead of TDID correlation #8

Open mike-shvets opened 5 years ago

mike-shvets commented 5 years ago

Core code is in attention.py. Config is updated with ATTENTION_ENABLED, ATTENTION_COMBINATION_MODE, and ATTENTION_NORMALIZE_MODE. TDID class is updated to optionally include attention.