Vérifier implémentation ENet en conformité avec publication

Rob174 / PIR

1 stars 0 forks source link

Closed Rob174 closed 3 years ago

Rob174 commented 3 years ago

Rob174 commented 3 years ago

Archi générale :

ENetArchi

Modules :

ENetblocks

Rob174 commented 3 years ago

Differences :

max_pool, indices = MaxPoolingWithArgmax2D()(inp) : dans la publication maxpooling simple
PReLU : shared axis [1,2](non précisés dans publication) : shared_axes doc

Encodeur ok

Decodeur normalement ok

Rob174 commented 3 years ago

TODO : revoir chargement des poids TODO : importer dans un script le modèle