Caja negra de red neuronal (Mobilenetv3): la extracción de la capa intermedia y la visualización del código de la capa intermedia a través de la pantalla CV2

Importar modelo de red neuronal 

 función de importación de modelo de red neuronal

1. A través del tensor de característica cíclica x, cada módulo en todo el modelo se extrae secuencialmente de acuerdo con el nombre de la capa intermedia y se almacena en la lista.

y=x[1:,:,:,:] 
d = y.permutar(0, 2, 3, 1).apretar(0) 

# cv2.imshow("y",y) 

y=d.numpy() # Aquí hay un método para convertir ndarray 
cv2.imshow("Crop", y) # str(name) necesita convertir una cadena 
cv2.waitKey(0) 
cv2.destroyAllWindows()

 

Convierta el tensor de características intermedias extraído a ndarray y luego ejecute cv2.imshow, solo consulte el código anterior