TimoSaemann / ENet

ENet: A Deep Neural Network Architecture for Real-Time Semantic Segmentation
577 stars 276 forks source link

ENet in Caffe2 #56

Open mmpinso opened 6 years ago

mmpinso commented 6 years ago

Hello, would it be possible to have ENet ready for Caffe2? ENet seems perfect for real time image segmentation on smartphones. Tensorflow implementation by Kwotsin cannot be deployed into CPU-only powered phones because of max_pooling_with_argmax kernel not being implemented for CPU by Tensorflow.