File size: 304 Bytes
19ea5c5
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
"""
Image captioning package: EfficientNetB0 encoder + GPT-2 decoder.

This package exposes the main components:
- ImageCaptioningModel (in model.py)
- dataset/dataloader utilities (in dataset.py)
- training, evaluation, and inference scripts.
"""

from .model import ImageCaptioningModel  # noqa: F401