Mjolnir65/FasterRCNN
0
1�
2 ���g� � �� � d dl Z d dlmZmZmZmZmZ d dlmZm Z d dl3m Z d dl
mZmZmZ d dlmZ d dlmZmZ d dlmZmZ G d � d4ej � � Z ed d � f�
� � e j dddd�de dee dedej f de deee dee defd�� � Z d dej de deee dee deedej f def d�Z de!d ee d e d e de f5d �Z"dS )!� N)�Callable�Dict�List�Optional�Union)�nn�Tensor)�misc)�
ExtraFPNBlock�FeaturePyramidNetwork�LastLevelMaxPool)�resnet)�_get_enum_from_fn� WeightsEnum)�handle_legacy_interface�IntermediateLayerGetterc � � � e Zd Z d
dej deeef dee dede e6 de e dej f d df� fd7�
Z d e
d eee
f fd �Z� xZS )�BackboneWithFPNN�backbone�
return_layers�in_channels_list� out_channels� extra_blocks�8norm_layer.�returnc �� �� t � � � � � |�t � � }t ||�� � | _ t ||||�� � | _ || _ d S )N)r )r r r r )�super�__init__r
r �bodyr �fpnr )�selfr r r r r r � __class__s ��W/home/gaurav/Documents/assignment-3-gaurav-behera/q1/modify/detection/backbone_utils.pyr zBackboneWithFPN.__init__ sr �� � �������� � �+�-�-�L�+�H�M�R�R�R�� �(� -�%�%�!� 9� 10� 11��� )����� �xc �Z � | � |� � }| � |� � }|S )N)r r )r! r% s r# �forwardzBackboneWithFPN.forward% s% � � �I�I�a�L�L�� �H�H�Q�K�K���r$ )NN)�__name__�12__module__� __qualname__r �Moduler �strr �intr r r r r r' �
__classcell__)r" s @r# r r
s� �� � � � � � 15�9=�)� )��)�)� �C��H�~�)� �s�)� )�13 � )� �}�-�
)� �X�c�2�9�n� 5�6�)� 14�)� )� )� )� )� )�.�� �D��f��$5� � � � � � � � r$ r �15pretrainedc �X � t t j | d � � d S )N�
backbone_name�
IMAGENET1K_V1)r r �__dict__)�kwargss r# �<lambda>r5 . s! � �(�� ���9P�)Q�R�R�Sb�c� r$ )�weights� )r �trainable_layers�returned_layersr r1 r6 r .r8 r9 r r c �^ � t j | ||�� � }t ||||� � S )N)r6 r )r r3 �_resnet_fpn_extractor)r1 r6 r r8 r9 r r s r# �resnet_fpn_backboner< + s2 � � ��}�-�g�*�U�U�U�H� ��+;�_�l� [� [�[r$ r c �P �16� � |dk s|dk rt d|� �� � �g d�d |� }|dk r|� d� � | � � � D ]5\ � }t � fd�|D � � � � r|� d� � �6|�t � � }|�g d�}t
|� � dk st |� � dk rt d |� �� � �d17� t |� � D � � }| j d z �18�19fd �|D � � }d
} t | ||| ||�� � S )Nr � z3Trainable layers should be in the range [0,5], got )�layer4�layer3�layer2�layer1�conv1�bn1c �< �� g | ]}�� |� � ��S � )�20startswith)�.0�layer�names �r# �21<listcomp>z)_resnet_fpn_extractor.<locals>.<listcomp>N s( �� �H�H�H�u�D�O�O�E�*�*�*�H�H�Hr$ F)� � r7 � z6Each returned layer should be in the range [1,4]. Got c �: � i | ]\ }}d |� �t |� � ��S )rI )r, )rH �v�ks r# �22<dictcomp>z)_resnet_fpn_extractor.<locals>.<dictcomp>X s* � �P�P�P�T�Q��[�Q�[�[�#�a�&�&�P�P�Pr$ � c �& �� g | ]
}�d |dz23 z z ��S )rM rL rF )rH �i�in_channels_stage2s �r# rK z)_resnet_fpn_extractor.<locals>.<listcomp>[ s' �� �S�S�S�a�*�Q�1�q�5�\�9�S�S�Sr$ � )r r ) �24ValueError�append�named_parameters�all�requires_grad_r
�min�max� enumerate�inplanesr ) r r8 r9 r r �layers_to_train� parameterr r r rV rJ s @@r# r; r; ? s� ��� � �!��� /�!� 3� 3��a�O_�a�a�b�b�b�G�G�G�HY�IY�HY�Z�O��1� � �� � �u�%�%�%�#�4�4�6�6� ,� ,���i� �H�H�H�H��H�H�H� I� I� ,� � $� $�U� +� +� +����'�)�)� �� �&�,�,��25�?���q� � �C��$8�$8�A�$=�$=��c�Ra�c�c�d�d�d�P�P�Y��5O�5O�P�P�P�M�!�*�a�/��S�S�S�S�?�S�S�S���L� ��-�!1�<�l�gq� � � � r$ �26is_trained�trainable_backbone_layers� max_value�
default_valuec � � | s |�t j d|� d�� � |}|�|}|dk s||k rt d|� d|� d�� � �|S )Nz�Changing trainable_backbone_layers has no effect if neither pretrained nor pretrained_backbone have been set to True, falling back to trainable_backbone_layers=z! so that all layers are trainabler z4Trainable backbone layers should be in the range [0,z], got � )�warnings�warnrX )rc rd re rf s r# �_validate_trainable_layersrk b s� � � � .� $� 0� �M�j�=F�j� j� j�
�
�
�27 %.�!� !�(�$1�!� �1�$�$�(A�I�(M�(M�� q�9� q� q�Un� q� q� q�28� 29� 30� %�$r$ )NNN)#ri �typingr r r r r �torchr r �torchvision.opsr31 � misc_nn_ops�'torchvision.ops.feature_pyramid_networkr r r
�torchvision.modelsr �torchvision.models._apir r �torchvision.models._utilsr r r+ r �FrozenBatchNorm2dr, r- r< �ResNetr; �boolrk rF r$ r# �<module>rw s� �� ���� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� � � � � � � � � /� /� /� /� /� /� j� j� j� j� j� j� j� j� j� j� %� %� %� %� %� %� B� B� B� B� B� B� B� B� V� V� V� V� V� V� V� V�� � � � �b�i� � � �< ���c�c�
�� � � ,7�+H� �+/�,0� \� \� \�� \� �k�
"� \� ��b�i� �(� \�32 � \� �d�3�i�(�
\� �=�)� \� � \� \� \�
� � \�" ,0�,0�59� � ��m� �� � �d�3�i�(� � �=�)� �33 ��#�r�y�.�1�2� � �
� � � �F%��%� '��}�%� �%� � %�34 � %� %� %� %� %� %r$