11
22# coding: utf-8
33
4- # In[5 ]:
4+ # In[46 ]:
55
66
77import MatchingModelPoints as match
1212import Image_preperation as prep
1313import PCA_analysis as PCA
1414import ActiveFitContour as af
15+ import time
1516
1617def active_shape_scale_n_times (img , tooth_points , pca_tooth , length , scale , n_times ):
1718
@@ -37,14 +38,22 @@ def active_shape_scale(img, tooth_points, pca_tooth, length, scale):
3738
3839 return new_scaled_points
3940
40- def active_shape (edge_img , tooth_points , pca_tooth , length , alfa ):
41+ def active_shape (edge_img , tooth_points , pca_tooth , length , alfa , activeFitON , MatchingON = True ):
4142
42- new_points , error = fit .fit_measure (tooth_points , length , edge_img )
43- # new_points = af.active_contour(tooth_points, edge_img, length, alfa)
44- b , pose_param = match .match_model_points (new_points , pca_tooth )
43+ if (activeFitON ):
44+ new_points = af .active_contour (tooth_points , edge_img , length , alfa )
45+ else :
46+ new_points , error = fit .fit_measure (tooth_points , length , edge_img )
47+ 48+ if (MatchingON ):
49+ b , pose_param = match .match_model_points (new_points , pca_tooth )
50+ x = match .generate_model_point (b , pca_tooth )
51+ y = match .inv_transform (x .reshape (40 ,2 ),pose_param )
52+ else :
53+ y = new_points
54+ 55+ return y
4556
46- x = match .generate_model_point (b , pca_tooth )
47- return match .inv_transform (x .reshape (40 ,2 ),pose_param )
4857
4958def active_shape_n_times (edge_img , tooth_points , pca_tooth , length ,alfa , n_times ):
5059
@@ -116,6 +125,13 @@ def testings():
116125 points_array = active_shape_n_times (edge_img , tooth , pca_tooth , 10 , 20 ,9 )
117126
118127 show_evolution (img , points_array )
128+ 129+ 130+ def show_with_points (img , points ):
131+ fig , ax = plt .subplots (figsize = (15 , 15 ))
132+ plt .imshow (img )
133+ plt .plot (points [:,0 ], points [:,1 ], 'ro' , markersize = 1 )
134+ plt .show ()
119135
120136
121137# In[2]:
@@ -140,5 +156,9 @@ def testings():
140156 plt .plot (tooth [:,0 ], tooth [:,1 ], 'ro' , markersize = 1 )
141157 plt .show ()
142158
143- points = active_shape_n_times (edge_img , tooth , pca_tooth , 10 , 1 ,1 )
159+ points_list = active_shape_n_times (edge_img , tooth , pca_tooth , 10 , 1 ,1 )
160+ 161+ #show_evolution(radiograph,points_list)
162+ for points in points_list :
163+ show_with_points (radiograph , points )
144164
0 commit comments