Exemplos de resnet152 em Python

Exemplo n.º 1

Arquivo: preprocess-images.py Projeto: LeeWilli/pytorch-vqa

    def __init__(self):
        super(Net, self).__init__()
        self.model = caffe_resnet.resnet152(pretrained=True)

        def save_output(module, input, output):
            self.buffer = output
        self.model.layer4.register_forward_hook(save_output)

Exemplo n.º 2

Arquivo: test_inference_with_heatmaps.py Projeto: christos-c/pytorch-vqa

    def __init__(self):
        super(ImgNet, self).__init__()
        self.model = caffe_resnet.resnet152(pretrained=True)

        # noinspection PyUnusedLocal
        def save_output(module, input, output):
            self.buffer = output
        self.model.layer4.register_forward_hook(save_output)