python VGG16影像轉CSV

woff

原圖：

python VGG16 影像轉CSV

結果圖：

python VGG16 影像轉CSV

完整代碼：

1. from keras.applications.vgg16 import VGG16
   
2. from keras.applications.vgg16 import preprocess_input
   
3. from keras.preprocessing.image import load_img
   
4. from keras.preprocessing.image import img_to_array
   
5. from keras.models import Model
   
6. from matplotlib import pyplot as plt
   
7. from numpy import expand_dims
   
8. model = VGG16()
   
9. model = Model(inputs=model.inputs, outputs=model.layers[2].output)
   
10. #summary 預設
    
11. model.summary()
    
12. # VGG16 只能用target_size=(224,224)
    
13. img = load_img('luffy.jpg',target_size=(224,224))
    
14. img = img_to_array(img)
    
15. print(img)
    
16. print(img.shape)
    
17. img = expand_dims(img, axis=0)
    
18. print(img)
    
19. print(img.shape)
    
20. img = preprocess_input(img)
    
21. feature_maps = model.predict(img)
    
22. square = 8
    
23. ix = 1
    
24. for _ in range(square):
    
25.     for _ in range(square):
    
26.         ax = plt.subplot(square,square,ix)
    
27.         ax.set_xticks([])
    
28.         ax.set_yticks([])
    
29.         plt.imshow(feature_maps[0, :, :, ix - 1], cmap='gray')
    
30.         ix += 1
    
31. plt.show()

複製代碼