lugiavn
July 2, 2019, 4:16am
1
2 years ago I was looking for a GoogLeNet model
Hey, I want to ask if anybody has the GoogLeNet model (trained on ImageNet)
I’m trying to reproduce some deep metric learning result and all papers use GoogLeNet, so for fair comparison and analysis…
Training the network myself or even converting it from tensorflow is too much work
Recently somebody has added a GoogLeNet model to torchvision
import warnings
from collections import namedtuple
from functools import partial
from typing import Any, Callable, List, Optional, Tuple
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch import Tensor
from ..transforms._presets import ImageClassification
from ..utils import _log_api_usage_once
from ._api import register_model, Weights, WeightsEnum
from ._meta import _IMAGENET_CATEGORIES
from ._utils import _ovewrite_named_param, handle_legacy_interface
__all__ = ["GoogLeNet", "GoogLeNetOutputs", "_GoogLeNetOutputs", "GoogLeNet_Weights", "googlenet"]
show original
The original paper is cited and also it says there that the model is ported from tensorflow.
1 Like