caffe.NetSpec
전언
caffe NetSpec 클래스
본문
caffe.NetSpec은 caffe/net 에 정의됨spec.py의 클래스 코드는 다음과 같습니다.
class NetSpec(object):
"""A NetSpec contains a set of Tops (assigned directly as attributes).
Calling NetSpec.to_proto generates a NetParameter containing all of the
layers needed to produce all of the assigned Tops, using the assigned
names.
"""
def __init__(self):
super(NetSpec, self).__setattr__('tops', OrderedDict())
def __setattr__(self, name, value):
self.tops[name] = value
def __getattr__(self, name):
return self.tops[name]
def __setitem__(self, key, value):
self.__setattr__(key, value)
def __getitem__(self, item):
return self.__getattr__(item)
def to_proto(self):
names = {v: k for k, v in six.iteritems(self.tops)}
autonames = Counter()
layers = OrderedDict()
for name, top in six.iteritems(self.tops):
top._to_proto(layers, names, autonames)
net = caffe_pb2.NetParameter()
net.layer.extend(layers.values())
return net
코드에서 설명: NetSpec은 Tops(직접 값을 속성으로 부여할 수 있음)를 포함하는 집합입니다.NetSpec을 호출합니다.to_proto는 모든 층 (layers) 을 포함하는 네트워크 파라미터를 만듭니다. 이 층 (layers) 은 값을 부여받아야 하고, 값을 부여받은 이름을 사용해야 합니다.
사용:
n=caffe.NetSpec () 은 Caffe를 가져오는 Net입니다. 이 n을 계속 채우기만 하면 맨 뒤에 n을 파일로 출력하면 Caffe에서 본 Net의protobuf의 정의를 배울 수 있습니다.
끝(^;^).
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
Windows 10의 Ubuntu에 Caffe를 배치하고 Single Shot MultiBox Detector 실행예전에는 Ubuntu on Windows 10에서 카페가 LMDB의 동작에 문제가 있었지만 지금은 동작의 상태인 것 같다.저는 버클리 대학의 카페입니다. Intel은 자체 CPU에서 빠른 컴퓨팅을 위한 컴파일러 및 ...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.