Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/affjljoo3581/Inverse-DALL-E-for-Optical-Character-Recognition
/ functions
Functions
61 in github.com/affjljoo3581/Inverse-DALL-E-for-Optical-Character-Recognition
⨍
Functions
61
◇
Types & classes
14
↓ 3 callers
Method
__init__
(self, input_dim: int, output_dim: int)
vqvae/src/modeling.py:9
↓ 2 callers
Method
create_attention_mask
(self, input_ids: list[int])
inverse-dalle/src/data.py:39
↓ 2 callers
Method
create_labels
(self, input_ids: list[int])
inverse-dalle/src/data.py:42
↓ 2 callers
Method
create_text_tokens
(self, text: str | None)
inverse-dalle/src/data.py:29
↓ 2 callers
Method
create_visual_tokens
(self, visual_tokens: str)
inverse-dalle/src/data.py:20
↓ 2 callers
Method
decode_to_text
(self, sequence: list[int])
inverse-dalle/src/lightning.py:86
↓ 2 callers
Method
resize_shape
(self, width: int, height: int)
vqvae/src/data.py:72
↓ 1 callers
Method
__init__
(self, config: DictConfig)
inverse-dalle/src/finetune.py:70
↓ 1 callers
Method
__init__
(self, config: DictConfig)
inverse-dalle/src/lightning.py:119
↓ 1 callers
Function
calculate_confidence
( sequence: list[int], sequence_index: int, scores: list[torch.Tensor], eos_token_id: int, )
inverse-dalle/src/predict.py:27
↓ 1 callers
Function
create_train_val_dataloaders
(config: DictConfig)
vqvae/src/data.py:112
↓ 1 callers
Function
decode_to_text
(sequence: list[int], vocab: list[str])
inverse-dalle/src/predict.py:16
↓ 1 callers
Function
main
(args: argparse.Namespace, config: DictConfig)
vqvae/src/predict.py:22
↓ 1 callers
Function
main
( name: str, config: DictConfig, resume_from: str | None = None, resume_id: str | None = None,
vqvae/src/train.py:18
↓ 1 callers
Function
main
(args: argparse.Namespace)
scripts/preprocess_external_dataset.py:14
↓ 1 callers
Function
main
(args: argparse.Namespace)
inverse-dalle/src/predict.py:44
↓ 1 callers
Function
main
( name: str, config: DictConfig, resume_from: str | None = None, resume_id: str | None = None,
inverse-dalle/src/train.py:18
↓ 1 callers
Function
main
( name: str, config: DictConfig, resume_from: str | None = None, resume_id: str | None = None,
inverse-dalle/src/finetune.py:94
↓ 1 callers
Function
main
(name: str, config: DictConfig, checkpoint: str)
inverse-dalle/src/export.py:14
↓ 1 callers
Method
parameter_groups
(self)
inverse-dalle/src/finetune.py:58
↓ 1 callers
Method
parameter_groups
(self)
inverse-dalle/src/lightning.py:103
Method
__getitem__
(self, index: int)
vqvae/src/data.py:27
Method
__getitem__
(self, index: int)
vqvae/src/data.py:76
Method
__getitem__
(self, index: int)
inverse-dalle/src/data.py:46
Method
__getitem__
(self, index: int)
inverse-dalle/src/data.py:68
Method
__init__
(self, config: DictConfig)
vqvae/src/lightning.py:22
Method
__init__
( self, num_channels: int = 3, num_layers: tuple[int, ...] = (2, 2, 2, 2, 2),
vqvae/src/modeling.py:25
Method
__init__
( self, num_channels: int = 3, num_layers: tuple[int, ...] = (2, 2, 2, 2, 2),
vqvae/src/modeling.py:55
Method
__init__
( self, num_embeddings: int = 16384, embedding_dim: int = 512, factorized_dim:
vqvae/src/modeling.py:85
Method
__init__
( self, image_files: list[str], max_length: int = 64, pooling: int = 16,
vqvae/src/data.py:50
Method
__init__
(self, dataset: ImageDatasetForTokenization, batch_size: int)
vqvae/src/data.py:94
Method
__init__
(self, config: DictConfig, vocab: list[str])
inverse-dalle/src/finetune.py:31
Method
__init__
(self, config: DictConfig, vocab: list[str])
inverse-dalle/src/lightning.py:24
Method
__iter__
(self)
vqvae/src/data.py:108
Method
__len__
(self)
vqvae/src/data.py:24
Method
__len__
(self)
vqvae/src/data.py:69
Method
__len__
(self)
vqvae/src/data.py:105
Method
__len__
(self)
inverse-dalle/src/data.py:17
Method
configure_optimizers
(self)
vqvae/src/lightning.py:77
Method
configure_optimizers
(self)
inverse-dalle/src/finetune.py:63
Method
configure_optimizers
(self)
inverse-dalle/src/lightning.py:108
Method
forward
( self, images: torch.Tensor )
vqvae/src/lightning.py:31
Method
forward
(self, hidden: torch.Tensor)
vqvae/src/modeling.py:17
Method
forward
(self, images: torch.Tensor)
vqvae/src/modeling.py:44
Method
forward
(self, latents: torch.Tensor)
vqvae/src/modeling.py:74
Method
forward
( self, encodings: torch.Tensor, quantize_only: bool = False )
vqvae/src/modeling.py:96
Method
forward
( self, labels: torch.Tensor, **kwargs: torch.Tensor )
inverse-dalle/src/lightning.py:37
Method
on_load_checkpoint
(self, checkpoint: dict[str, Any])
vqvae/src/lightning.py:80
Method
on_load_checkpoint
(self, checkpoint: dict[str, Any])
inverse-dalle/src/lightning.py:113
Method
setup
(self, stage: Optional[str] = None)
inverse-dalle/src/finetune.py:76
Method
setup
(self, stage: Optional[str] = None)
inverse-dalle/src/lightning.py:129
Method
train_dataloader
(self)
inverse-dalle/src/finetune.py:84
Method
train_dataloader
(self)
inverse-dalle/src/lightning.py:143
Method
training_step
(self, images: torch.Tensor, batch_idx: int)
vqvae/src/lightning.py:50
Method
training_step
(self, batch: dict[str, torch.Tensor], idx: int)
inverse-dalle/src/finetune.py:38
Method
training_step
(self, batch: dict[str, torch.Tensor], idx: int)
inverse-dalle/src/lightning.py:56
Method
val_dataloader
(self)
inverse-dalle/src/lightning.py:152
Method
validation_epoch_end
(self, outputs: tuple[torch.Tensor, torch.Tensor] | None)
vqvae/src/lightning.py:69
Method
validation_epoch_end
(self, outputs: list[list[tuple[str, str]]])
inverse-dalle/src/lightning.py:96
Method
validation_step
( self, images: torch.Tensor, batch_idx: int )
vqvae/src/lightning.py:56
Method
validation_step
( self, batch: dict[str, torch.Tensor], idx: int )
inverse-dalle/src/lightning.py:62