TransVW icon indicating copy to clipboard operation
TransVW copied to clipboard

Question about Auto-Encoder as Feature extractor

Open babbu3682 opened this issue 3 years ago • 1 comments

I read your paper but you said you used auto-encoder as a feature extractor.

However, as I analyze your code, you use U-net architecture or V-net architecture. which one is right?

I think Auto-encoder is more reasonable because U-net can cheat by using skip connection.

babbu3682 avatar Nov 05 '21 12:11 babbu3682

In general, any pre-trained network can be used as the feature extractor. The train_autoencoder.py is parametric; for Auto-encoder, the option --arch Vnet should be used. The network can be trained with other restoration tasks with U-Net as well.

fhaghighi avatar Nov 19 '21 02:11 fhaghighi