Возможно ли добавить новый слой (для другой потери) с помощью Caffe? Как это сделать, используя Matlab или Python?Можно ли добавить новые слои в Caffe?
1
A
ответ
2
Да, вы можете добавить пользовательскую функцию потерь с помощью pycaffe. Вот пример слоя потерь евклидов в питоне (взято из Caffe Github repo). Вы должны предоставить функцию потерь в прямой функции, и это градиент в обратном методе:
import caffe
import numpy as np
class EuclideanLossLayer(caffe.Layer):
"""
Compute the Euclidean Loss in the same manner as the C++ EuclideanLossLayer
to demonstrate the class interface for developing layers in Python.
"""
def setup(self, bottom, top):
# check input pair
if len(bottom) != 2:
raise Exception("Need two inputs to compute distance.")
def reshape(self, bottom, top):
# check input dimensions match
if bottom[0].count != bottom[1].count:
raise Exception("Inputs must have the same dimension.")
# difference is shape of inputs
self.diff = np.zeros_like(bottom[0].data, dtype=np.float32)
# loss output is scalar
top[0].reshape(1)
def forward(self, bottom, top):
self.diff[...] = bottom[0].data - bottom[1].data
top[0].data[...] = np.sum(self.diff**2)/bottom[0].num/2.
def backward(self, top, propagate_down, bottom):
for i in range(2):
if not propagate_down[i]:
continue
if i == 0:
sign = 1
else:
sign = -1
bottom[i].diff[...] = sign * self.diff/bottom[i].num
Сохранить это как pyloss.py, например. Затем вы можете использовать питона слой в файле prototxt, чтобы загрузить его:
layer {
type: 'Python'
name: 'loss'
top: 'loss'
bottom: 'ipx'
bottom: 'ipy'
python_param {
# the module name -- usually the filename -- that needs to be in $PYTHONPATH
module: 'pyloss'
# the layer name -- the class name in the module
layer: 'EuclideanLossLayer'
}
# set loss weight so Caffe knows this is a loss layer.
# since PythonLayer inherits directly from Layer, this isn't automatically
# known to Caffe
loss_weight: 1
}
Или в вашем питона сценария:
n.loss = L.Python(n.ipx, n.ipy,python_param=dict(module='pyloss',layer='EuclideanLossLayer'),
loss_weight=1)
Просто быть очень осторожным в вычислениях и implemeting градиент (обратная функция), так как он, как правило, подвержен ошибкам.
Смежные вопросы
- 1. Можно ли добавить новые функции в PHP?
- 2. Можно ли добавить слои в .tiff изображения с помощью .NET?
- 3. Geodjango OSMGeoAdmin: как можно добавить разные слои
- 4. Все новые слои si прозрачные
- 5. Добавить слои данных в matplotlib
- 6. Можно ли добавить новые столбцы в DataFrame в Pandas (python)?
- 7. Можно ли добавить новые треки в плейлист для воспроизведения?
- 8. Можно ли добавить новые поля в ссылку Link Link
- 9. В keras предварительно подготовленных моделях, можно ли удалить слои?
- 10. Можно ли расширить Flask-беспокойный, чтобы добавить новые маршруты?
- 11. Как добавить слои в карты?
- 12. Как добавить символы в слои?
- 13. Можно ли создать новые теги в Мако?
- 14. Преобразование результата TensorRT в Caffe
- 15. Имеет ли Caffe статическое фиаско инициализации?
- 16. Соединились ли слои Magento?
- 17. cocos2d добавить слои на сцену
- 18. Добавить новые иконки в OpenUI5?
- 19. Как добавить слои Convolution2D? Theano
- 20. C SDL2- можно ли устранить некоторые слои абстракции?
- 21. Как добавить слои на просмотр?
- 22. Python: Можно ли добавлять новые методы в класс Tuple
- 23. Слои изображения в CSS
- 24. Добавить правильно размещенные слои в контекст
- 25. Новые композиционные слои в Google Chrome из-за unsquashable содержание
- 26. Можно ли добавить ._ * в gitignore?
- 27. Caffe: установить модифицированные проекты Caffe на Windows
- 28. Можно ли добавить триггеры?
- 29. Добавить слои в PSD-файл, используя ImageMagick
- 30. Как добавить слои чертежа в размере карт
вы можете использовать [эту реализацию] (https://github.com/tnarihi/caffe/blob/python-gradient-checker/python/caffe/test/test_gradient_checker.py) утилиты проверки градиента для python. – Shai