Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training on DEM SRTM dataset #451

Open
htn274 opened this issue Jun 29, 2020 · 2 comments
Open

Training on DEM SRTM dataset #451

htn274 opened this issue Jun 29, 2020 · 2 comments

Comments

@htn274
Copy link

@htn274 htn274 commented Jun 29, 2020

Thank you for your great work. I have been applying your research to my problem now.
Here is my problem:

DEM, stands for Digital Elevation Model, is a numerical matrix with each pixel represents its elevation correspond to a location.

SRTM is a globally DEM dataset. However, with a reservoir or a lake, SRTM showed the elevation of surface water at the time it collected (in 2000). Now I want to recover DEM below surface water of a lake or a reservoir.

Here is my step-by-step process:

  1. My training set is 15k DEM images.

  2. I generated masks correspond to input image. An input image has a specific mask. My mask is an connected area which has elevation lower than a random number. So it can describe a reservoir or a lake, I think so. Some of my masks with the input:

image

So I customized your code to read my input:
image

image

  1. I also customized data_from_fnames in neuralgym toolkit to read '.tif' files. Then I used min-max normalize DEM image to 0-255 in order to pass it to your model.

  2. Some first rows in my train.flist:

image

  1. Because of limit memory of my GPU, I set image input shape (128, 128, 1) and batch size is 16 as your advice I saw in other issues.
#training
train_spe: 1000 
max_iters: 1000000
viz_max_out: 10
val_psteps: 500

static_view_size: 30
img_shapes: [128, 128, 1]
height: 128
width: 128
max_delta_height: 32
max_delta_width: 32
batch_size: 16
vertical_margin: 0
horizontal_margin: 0
  1. I finally trained more than 70 epoches and get bad results:

My losses are not converging

image

image

Some of generated validation images:

image

Questions

  1. Are all my customizations correct?
  2. I saw your sample flist is shuffled and I didn't. Does it affect to the training result?
  3. Could you give me some suggestion and your views on my problem?

Thanks a lot! <3

@Efrat-Taig
Copy link

@Efrat-Taig Efrat-Taig commented Jul 13, 2020

Hi htn274
This seems like a nice idea, how did it work out for you?
I am trying to implement a similar idea, with non random masks, and it can be very helpful to learn from your experience.

Thanks in advance

@JiahuiYu
Copy link
Owner

@JiahuiYu JiahuiYu commented Aug 13, 2020

@htn274 Thanks for your detailed feedback and that's exactly our community needs! I will keep this issue in the front page.

Regarding your question:

  1. I think you are right, and the results are good already?
  2. Shuffling should always be non-worse.
  3. Larger dataset could help.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Linked pull requests

Successfully merging a pull request may close this issue.

None yet
3 participants
You can’t perform that action at this time.