Browse Source

nomal debug finish!

liuyuqi 7 years ago
parent
commit
3279d8d8f3
5 changed files with 62 additions and 54 deletions
  1. 3 0
      .gitignore
  2. 7 4
      src/python/main.py
  3. 19 0
      src/python/plot1.py
  4. 31 13
      src/python/plot2.py
  5. 2 37
      src/python/result.py

+ 3 - 0
.gitignore

@@ -0,0 +1,3 @@
+/output
+/.vscode
+/input

+ 7 - 4
src/python/main.py

@@ -13,10 +13,13 @@ import tifffile as tiff
 import matplotlib.pyplot as plt
 from matplotlib import cm
 
-FILE_2015 = '/media/sf_share/linux/workspaces/guangdong/input/quickbird2015.tif'
-FILE_2017 = '/media/sf_share/linux/workspaces/guangdong/input/quickbird2017.tif'
-FILE_cadastral2015 = '/media/sf_share/linux/workspaces/guangdong/input/cadastral2015.tif'
-FILE_tinysample = '/media/sf_share/linux/workspaces/guangdong/input/tinysample.tif'
+getcwd()
+chdir("/media/sf_share/linux/workspaces/guangdong")
+
+FILE_2015 = './input/quickbird2015.tif'
+FILE_2017 = './input/quickbird2017.tif'
+FILE_cadastral2015 = './input/cadastral2015.tif'
+FILE_tinysample = './input/tinysample.tif'
 
 #read images
 im_2015 = tiff.imread(FILE_2015).transpose([1, 2, 0])

+ 19 - 0
src/python/plot1.py

@@ -0,0 +1,19 @@
+fig, axes = plt.subplots(ncols=2, nrows=2, figsize=(14, 10))
+
+p1 = plt.subplot(221)
+i1 = p1.imshow(im_2015[100:1000, 100:1000, 0])
+plt.colorbar(i1)
+
+p2 = plt.subplot(222)
+i2 = p2.imshow(im_2015[100:1000, 100:1000, 1])
+plt.colorbar(i2)
+
+p3 = plt.subplot(223)
+i3 = p3.imshow(im_2015[100:1000, 100:1000, 2])
+plt.colorbar(i3)
+
+p4 = plt.subplot(224)
+i4 = p4.imshow(im_2015[100:1000, 100:1000, 3])
+plt.colorbar(i4)
+
+plt.show()

+ 31 - 13
src/python/plot2.py

@@ -1,19 +1,37 @@
-fig, axes = plt.subplots(ncols=2, nrows=2, figsize=(14, 10))
+x_start = 500
+x_end = 1000
+y_start = 4800
+y_end = 5300
 
-p1 = plt.subplot(221)
-i1 = p1.imshow(im_2015[100:1000, 100:1000, 0])
-plt.colorbar(i1)
+plt.subplots(ncols=3, nrows=2, figsize=(16, 8))
 
-p2 = plt.subplot(222)
-i2 = p2.imshow(im_2015[100:1000, 100:1000, 1])
-plt.colorbar(i2)
+mt = np.ma.array(np.ones((x_end-x_start, y_end-y_start)), 
+                 mask=((im_tiny[x_start:x_end, y_start:y_end, 0]/np.max(im_tiny)+im_cada[x_start:x_end, y_start:y_end])==0))
 
-p3 = plt.subplot(223)
-i3 = p3.imshow(im_2015[100:1000, 100:1000, 2])
-plt.colorbar(i3)
+p151 = plt.subplot(231)
+i151 = p151.imshow(im_2015[x_start:x_end, y_start:y_end, 3])
+plt.colorbar(i151)
 
-p4 = plt.subplot(224)
-i4 = p4.imshow(im_2015[100:1000, 100:1000, 3])
-plt.colorbar(i4)
+p152 = plt.subplot(233)
+i152 = p152.imshow(im_tiny[x_start:x_end, y_start:y_end, 0])
+plt.colorbar(i152)
+
+p153 = plt.subplot(232)
+i153 = p153.imshow(im_2015[x_start:x_end, y_start:y_end, 3])
+p153.imshow(mt, cmap=cm.bwr, alpha=0.3, vmin=0, vmax=1)
+plt.colorbar(i153)
+
+p171 = plt.subplot(234)
+i171 = p171.imshow(im_2017[x_start:x_end, y_start:y_end, 3])
+plt.colorbar(i171)
+
+p172 = plt.subplot(236)
+i172 = p172.imshow(im_cada[x_start:x_end, y_start:y_end])
+plt.colorbar(i172)
+
+p173 = plt.subplot(235)
+i173 = p173.imshow(im_2017[x_start:x_end, y_start:y_end, 3])
+p173.imshow(mt, cmap=cm.bwr, alpha=0.3, vmin=0, vmax=1)
+plt.colorbar(i173)
 
 plt.show()

+ 2 - 37
src/python/result.py

@@ -1,37 +1,2 @@
-x_start = 500
-x_end = 1000
-y_start = 4800
-y_end = 5300
-
-plt.subplots(ncols=3, nrows=2, figsize=(16, 8))
-
-mt = np.ma.array(np.ones((x_end-x_start, y_end-y_start)), 
-                 mask=((im_tiny[x_start:x_end, y_start:y_end, 0]/np.max(im_tiny)+im_cada[x_start:x_end, y_start:y_end])==0))
-
-p151 = plt.subplot(231)
-i151 = p151.imshow(im_2015[x_start:x_end, y_start:y_end, 3])
-plt.colorbar(i151)
-
-p152 = plt.subplot(233)
-i152 = p152.imshow(im_tiny[x_start:x_end, y_start:y_end, 0])
-plt.colorbar(i152)
-
-p153 = plt.subplot(232)
-i153 = p153.imshow(im_2015[x_start:x_end, y_start:y_end, 3])
-p153.imshow(mt, cmap=cm.bwr, alpha=0.3, vmin=0, vmax=1)
-plt.colorbar(i153)
-
-p171 = plt.subplot(234)
-i171 = p171.imshow(im_2017[x_start:x_end, y_start:y_end, 3])
-plt.colorbar(i171)
-
-p172 = plt.subplot(236)
-i172 = p172.imshow(im_cada[x_start:x_end, y_start:y_end])
-plt.colorbar(i172)
-
-p173 = plt.subplot(235)
-i173 = p173.imshow(im_2017[x_start:x_end, y_start:y_end, 3])
-p173.imshow(mt, cmap=cm.bwr, alpha=0.3, vmin=0, vmax=1)
-plt.colorbar(i173)
-
-plt.show()
+result_temp = np.random.randint(2, size=im_cada.shape, dtype=np.uint8)
+tiff.imsave('./output/result_temp.tiff', result_temp)