\begin{figure}[htbp]
\centering 
    \includegraphics[width=\textwidth]{./images/u-Net Diagram Representation.pdf}
  \caption{Representation of a 3D U-Net, the second network investigated for AM generation with eye-gaze supervision. %The left part acts as feature extractor, while the right part takes the abstract representation to create the semantic attention map. The skip connections provides additional information that helps the right part in generating semantic features
  }
  \label{fig: u-net diagram}
\end{figure}