search

ucaszyp / STEPS

This is the official repository for ICRA-2023 paper "STEPS: Joint Self-supervised Nighttime Image Enhancement and Depth Estimation"
https://arxiv.org/abs/2302.01334
MIT License
175 stars 10 forks source link

Is this model pre-trained with ImageNet? #14

Open breknddone opened 8 months ago

breknddone commented 8 months ago

Is this model pre-trained with ImageNet?Is the purpose of first training using the daytime data set just to get the pre-trained weights for night training?Thanks！

ucaszyp commented 8 months ago

We used the pretrained ResNet models and the daytime pretraining is used to train patch-GAN.

breknddone commented 8 months ago

We used the pretrained ResNet models and the daytime pretraining is used to train patch-GAN.

So can it be understood that the pre-trained weights during the day and night are both Resnet models? Is the training during the day so that Patch-GAN can work in the next step?