search

Maartendrinhuyzen / ai4mi_project

MIT License
0 stars 0 forks source link

Add attention mechanism to Unet (or Enet) #15

Closed Maartendrinhuyzen closed 2 weeks ago

Maartendrinhuyzen commented 2 weeks ago

I added a file called UnetAttention.py that is just Unet like Richter implemented but then with an attention mechanism added to it

Maartendrinhuyzen commented 2 weeks ago

Just pushed the latest version of the added attention mechanism. However I have not tested its performance yet. TBD.