Functions — github.com/AmrMaghraby/CNN-RNN-A-Unified-Framework-for-Multi-label-Image-Classification

hub / github.com/AmrMaghraby/CNN-RNN-A-Unified-Framework-for-Multi-label-Image-Classification / functions

Functions17 in github.com/AmrMaghraby/CNN-RNN-A-Unified-Framework-for-Multi-label-Image-Classification

⨍ Functions 17 ◇ Types & classes 4

↓ 5 callersMethodadd_word

(self, word)

build_vocab.py:14

↓ 1 callersMethod__init__

Load the pretrained ResNet-152 and replace top fc layer.

model.py:8

↓ 1 callersFunctionbuild_vocab

(json, threshold)

build_vocab.py:28

↓ 1 callersFunctionget_loader

Returns torch.utils.data.DataLoader for custom coco dataset.

data_loader.py:23

↓ 1 callersFunctionload_image

(image_path, transform=None)

sample.py:16

↓ 1 callersFunctionmain

(args)

train.py:17

↓ 1 callersFunctionmain

(args)

build_vocab.py:44

↓ 1 callersFunctionmain

(args)

sample.py:25

↓ 1 callersMethodsample

Generate captions for given image features using greedy search.

(self, root, json, vocab, transform=None)

data_loader.py:14

Method__init__

Set the hyper-parameters and build the layers.

Extract feature vectors from input images.

model.py:17

Methodforward

Decode image feature vectors and generates captions.

model.py:35