search

jhjacobsen / pytorch-i-revnet

i-RevNet Pytorch Code
MIT License
390 stars 46 forks source link

Questions about using the i-resnet for other applications. #23

Open xyIsHere opened 3 years ago

xyIsHere commented 3 years ago

Dear authors,

Thank you for your great work. Currently, I'm working on the application of semantic segmentation. So I wonder whether the i-resnet can be directly applied by semantic segmentation by simply changing the final classification layer (i.e. modified the following four lines). Or do u have other suggestions?

https://github.com/jhjacobsen/pytorch-i-revnet/blob/307413043e33540cbe9c3746ef420261f8138315/models/iRevNet.py#L134-L137

Thank you very much for your help in advance.

GewelsJI commented 2 years ago

Hi, @xyIsHere

Do you have any progress on semantic segmentation?