search

warmspringwinds / pytorch-segmentation-detection

Image Segmentation and Object Detection in Pytorch
MIT License
746 stars 170 forks source link

about Resnet18_8s #1

Closed MahlerMozart closed 7 years ago

MahlerMozart commented 7 years ago

Hello, I am very impressed by your great work! However, I am a little confused when I look at your Resnet18_8s network. I assume Resnet18_8s follows your approach in your paper "Deep Residual Learning for Instrument Segmentation in Robotic Surgery", which employ dilated convolutions to keep resolution. But in resnet_dilated.py, I could not find any dilated convolutions in Class Resnet18_8s. Could you please give more detailed explanation on the structure of Resnet18_8s? Many thanks.

warmspringwinds commented 7 years ago

Hi, @MahlerMozart .

For this repository we use a custom fork of pytorch/vision where we add dilated convolutions.

Look here: https://github.com/warmspringwinds/vision/blob/fully_conv_resnet/torchvision/models/resnet.py

I will update the front page Readme to better show that.

MahlerMozart commented 7 years ago

@warmspringwinds Perfect explanation. Thanks!