Ryanfafa's picture
Upload 7 files
19ea5c5 verified
raw
history blame contribute delete
304 Bytes
"""
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