






# coding=utf-8
# Copyright 2023 The HuggingFace Inc. team.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# this script dumps information about the environment

import os
import platform
import sys


__A  			=					"""3"""

print('Python version:', sys.version)

print('OS platform:', platform.platform())
print('OS architecture:', platform.machine())

try:
				import torch

				print('Torch version:', torch.__version__)
				print('Cuda available:', torch.cuda.is_available())
				print('Cuda version:', torch.version.cuda)
				print('CuDNN version:', torch.backends.cudnn.version())
				print('Number of GPUs available:', torch.cuda.device_count())
except ImportError:
				print('Torch version:', None)

try:
				import transformers

				print('transformers version:', transformers.__version__)
except ImportError:
				print('transformers version:', None)








import json
import os
from collections import Counter

import torch
import torchvision
import torchvision.transforms as transforms
from PIL import Image
from torch import nn
from torch.utils.data import Dataset


_lowerCamelCase    :		Dict		      =    {1: (1, 1), 2: (2, 1), 3: (3, 1), 4: (2, 2), 5: (5, 1), 6: (3, 2), 7: (7, 1), 8: (4, 2), 9: (3, 3)}


class 						lowercase    (							nn.Module):
						'''simple docstring'''


						def __init__(  self    :				Optional[Any]       ,	snake_case    :				int	):



											'''simple docstring'''


											super().__init__()
											SCREAMING_SNAKE_CASE							:     Optional[Any]						=				torchvision.models.resnetaaa(pretrained=snake_case	)
											SCREAMING_SNAKE_CASE							:     Optional[Any]						=				list(model.children()	)[:-2]
											SCREAMING_SNAKE_CASE							:     List[str]						=				nn.Sequential(*snake_case	)
											SCREAMING_SNAKE_CASE							:     Optional[int]						=				nn.AdaptiveAvgPoolad(POOLING_BREAKDOWN[args.num_image_embeds]	)







						def       lowerCamelCase_				(  self    :				Any       ,	snake_case    :				Optional[Any]	):



											'''simple docstring'''


											SCREAMING_SNAKE_CASE							:     Optional[int]						=				self.pool(self.model(snake_case	)	)
											SCREAMING_SNAKE_CASE							:     Any						=				torch.flatten(snake_case       ,	start_dim=2	)
											SCREAMING_SNAKE_CASE							:     Union[str, Any]						=				out.transpose(1       ,	2	).contiguous()
											return out  # BxNx2048


class 						lowercase    (							SCREAMING_SNAKE_CASE_):
						'''simple docstring'''


						def __init__(  self    :				str       ,	snake_case    :				Optional[Any]       ,	snake_case    :				Union[str, Any]       ,	snake_case    :				Optional[int]       ,	snake_case    :				int       ,	snake_case    :				Optional[Any]	):



											'''simple docstring'''


											SCREAMING_SNAKE_CASE							:     Optional[Any]						=				[json.loads(snake_case	) for l in open(snake_case	)]
											SCREAMING_SNAKE_CASE							:     Optional[Any]						=				os.path.dirname(snake_case	)
											SCREAMING_SNAKE_CASE							:     Optional[int]						=				tokenizer
											SCREAMING_SNAKE_CASE							:     Dict						=				labels
											SCREAMING_SNAKE_CASE							:     Optional[Any]						=				len(snake_case	)
											SCREAMING_SNAKE_CASE							:     int						=				max_seq_length

											SCREAMING_SNAKE_CASE							:     List[Any]						=				transforms







						def __len__(  self    :				str	):



											'''simple docstring'''


											return len(self.data	)







						def __getitem__(  self    :				Dict       ,	snake_case    :				int	):



											'''simple docstring'''


											SCREAMING_SNAKE_CASE							:     Optional[int]						=				torch.LongTensor(self.tokenizer.encode(self.data[index]['text']       ,	add_special_tokens=snake_case	)	)
											SCREAMING_SNAKE_CASE					,  SCREAMING_SNAKE_CASE					,  SCREAMING_SNAKE_CASE							:     Optional[Any]						=				sentence[0], sentence[1:-1], sentence[-1]
											SCREAMING_SNAKE_CASE							:     str						=				sentence[: self.max_seq_length]

											SCREAMING_SNAKE_CASE							:     Optional[Any]						=				torch.zeros(self.n_classes	)
											SCREAMING_SNAKE_CASE							:     Optional[int]						=				1

											SCREAMING_SNAKE_CASE							:     Optional[Any]						=				Image.open(os.path.join(self.data_dir       ,	self.data[index]['img']	)	).convert('RGB'	)
											SCREAMING_SNAKE_CASE							:     Tuple						=				self.transforms(snake_case	)

											return {
											    "image_start_token": start_token,
											    "image_end_token": end_token,
											    "sentence": sentence,
											    "image": image,
											    "label": label,
											}







						def       lowerCamelCase_				(  self    :				Any	):



											'''simple docstring'''


											SCREAMING_SNAKE_CASE							:     int						=				Counter()
											for row in self.data:
																label_freqs.update(row['label']	)
											return label_freqs


def        __a			(   __lowerCAmelCase   )     ->     Union[str, Any]:
					SCREAMING_SNAKE_CASE							:     str						=				[len(row['sentence']   ) for row in batch]
					SCREAMING_SNAKE_CASE					,  SCREAMING_SNAKE_CASE							:     Dict						=				len(__lowerCAmelCase   ), max(__lowerCAmelCase   )

					SCREAMING_SNAKE_CASE							:     List[Any]						=				torch.zeros(__lowerCAmelCase      ,	__lowerCAmelCase      ,	dtype=torch.long   )
					SCREAMING_SNAKE_CASE							:     List[str]						=				torch.zeros(__lowerCAmelCase      ,	__lowerCAmelCase      ,	dtype=torch.long   )

					for i_batch, (input_row, length) in enumerate(zip(__lowerCAmelCase      ,	__lowerCAmelCase   )   ):
										SCREAMING_SNAKE_CASE							:     Optional[Any]						=				input_row['sentence']
										SCREAMING_SNAKE_CASE							:     Optional[Any]						=				1

					SCREAMING_SNAKE_CASE							:     List[str]						=				torch.stack([row['image'] for row in batch]   )
					SCREAMING_SNAKE_CASE							:     Union[str, Any]						=				torch.stack([row['label'] for row in batch]   )
					SCREAMING_SNAKE_CASE							:     Tuple						=				torch.stack([row['image_start_token'] for row in batch]   )
					SCREAMING_SNAKE_CASE							:     Any						=				torch.stack([row['image_end_token'] for row in batch]   )

					return text_tensor, mask_tensor, img_tensor, img_start_token, img_end_token, tgt_tensor

def        __a			(   )     ->     str:
					return [
					    "Crime",
					    "Drama",
					    "Thriller",
					    "Action",
					    "Comedy",
					    "Romance",
					    "Documentary",
					    "Short",
					    "Mystery",
					    "History",
					    "Family",
					    "Adventure",
					    "Fantasy",
					    "Sci-Fi",
					    "Western",
					    "Horror",
					    "Sport",
					    "War",
					    "Music",
					    "Musical",
					    "Animation",
					    "Biography",
					    "Film-Noir",
					]



def        __a			(   )     ->     Union[str, Any]:
					return transforms.Compose(
					    [
					        transforms.Resize(256   ),
					        transforms.CenterCrop(224   ),
					        transforms.ToTensor(),
					        transforms.Normalize(
					            mean=[0.46_777_044, 0.44_531_429, 0.40_661_017]      ,	std=[0.12_221_994, 0.12_145_835, 0.14_380_469]      ,	),
					    ]   )