GirlsINSAT/Projet_AIF_HF
1
1o
2 ���i
� @ s< d dl Z d dlmZ d dlm mZ G dd� dej�ZdS )� Nc s. e Zd Zd� fdd� Zdd� Zdd� Z� ZS ) �MovieposterNet�3 c s� t t| ��� tjddddd�| _t�d�| _tjddddd�| _t�d�| _ tjddddd�| _4t�d�| _ tjddddd�| _ t�d�| _
t�dd�| _t�d d5�| _t�d �| _t�d6|�| _d S ) N� � � )� kernel_size�padding� �@ � � i b i g �?)�superr �__init__�nn�Conv2d�conv1� BatchNorm2d�bn1�conv2�bn2�conv3�bn3�conv4�bn4� MaxPool2d�pool�Linear�fc1�Dropout�dropout�fc2)�self� num_classes�� __class__� �4C:\Users\Admin\Documents\insa\5A\projet_AIF\model.pyr s zMovieposterNet.__init__c C s� | � t�| �| �|����}| � t�| �| �|����}| � t�| �| �|����}| � t�| � | �7|����}t � |d�}t�| �
|��}| �|�}| �|�}|S �Nr )r �F�relur r r r r r r r �torch�flattenr r r �r! �xr% r% r& �forward" s 89zMovieposterNet.forwardc C s| | � t�| �| �|����}| � t�| �| �|����}| � t�| �| �|����}| � t�| � | �10|����}t � |d�S r' )
r r( r) r r r r r r r r r* r+ r, r% r% r& � get_features. s11 zMovieposterNet.get_features)r )�__name__�12__module__� __qualname__r r. r/ �
__classcell__r% r% r# r& r s r )r* �torch.nnr �torch.nn.functional�13functionalr( �Moduler r% r% r% r& �<module> s