Morten Hjorth-Jensen
31 lines
667 B
Python
31 lines
667 B
Python
import numpy as np
|
|
from imageio import imread
|
|
import matplotlib.pyplot as plt
|
|
from mpl_toolkits.mplot3d import Axes3D
|
|
from matplotlib import cm
|
|
import numpy as np
|
|
from numpy.random import normal, uniform
|
|
|
|
# Load the terrain
|
|
terrain = imread('SRTM_data_Norway_1.tif')
|
|
|
|
N = 1000
|
|
m = 5 # polynomial order
|
|
terrain = terrain[:N,:N]
|
|
# Creates mesh of image pixels
|
|
x = np.linspace(0,1, np.shape(terrain)[0])
|
|
y = np.linspace(0,1, np.shape(terrain)[1])
|
|
x_mesh, y_mesh = np.meshgrid(x,y)
|
|
|
|
z = terrain
|
|
X = create_X(x_mesh, y_mesh,m)
|
|
|
|
|
|
# Show the terrain
|
|
plt.figure()
|
|
plt.title('Terrain over Norway 1')
|
|
plt.imshow(terrain, cmap='gray')
|
|
plt.xlabel('X')
|
|
plt.ylabel('Y')
|
|
plt.show()
|