krisrs1128 / clouds_dist

Simulation of low-clouds, from weather measures.
4 stars 1 forks source link

the metos selected in cpu_images are hard-coded #163

Open jassiene-mila opened 4 years ago

jassiene-mila commented 4 years ago

At this line, dimensions (metos) [22,25) are selected. The low clouds dataset has only 8 metos so it will cause an error. Besides, the tensor resulting from to_0_1(input_tensor[i, 22:25]) will have a shape (3,img_width,img_height). Images in the low_clouds dataset have only 1 channel so the concatenation on the last (-1) axis will fail (cannot concat (3,x,y) with (1,x,y) on axis -1)

krisrs1128 commented 4 years ago

@jassiene-mila I'm guessing you already made the change for this? If so, please push that branch, and we can review / merge.