davisking/dlib

DNN Support for dilated convolutions

開放

#1,018 建立於 2017年12月17日

 (15 則留言) (0 個反應) (0 位負責人)C++ (3,442 個分叉)batch import
help wanted

倉庫指標

星標
 (14,434 顆星)
PR 合併指標
 (平均合併 3小時 56分鐘) (30 天內合併 2 個 PR)

描述

Hello all! First I want to send my gratitude to all the people working on this great library. It is one of my favourite ML/image processing libraries out there. I specially like the DNN part of it it is so much easier to use then some of the others out there and that you can drive it directly in C++ is great! However I did notice there are some holes in functionality and I was wonder in particular if there is any future plan to add dilated convolutions option to the current DNN convolution layer. I hacked in some support myself to see if they do make a difference so I changed the dnn_segmentation example to add them. Unfortunately my hack would require all examples to be rewritten as there are now two additional parameters in the con layer.

Here are some example images using a trained head segmentation dataset testing it on some celebA images with and without dilation and you can see some decent improvements by just using dilation (that's the only change I didn't touch the network structure itself so far)

Non dilated example 1 Dilated example 1 Non dilated example2 Dilated example 2 Non dilated example3 Dilated example 3 Non dilated example4 Dilated example 4

貢獻者指南