Advertisement
MARSHAL327

Untitled

Jan 8th, 2024
115
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.83 KB | None | 0 0
  1. def three_layer_convnet(x, params):
  2.  
  3. """
  4. Трехслойная сверточная сеть с описанной выше архитектурой.
  5.  
  6. Входы:
  7. - x: тензор формы (N, H, W, 3), представляющий мини-блок изображений
  8. - params: список тензоров, представляющих веса и смещения
  9. сети; должен содержать следующее:
  10. - conv_w1: тензор формы (KH1, KW1, 3, channel_1) -
  11. веса первого сверточного слоя.
  12. - conv_b1: тензор формы (channel_1,) - смещения
  13. первого сверточного слоя.
  14. - conv_w2: тензор формы (KH2, KW2, channel_1, channel_2) -
  15. веса второго сверточного слоя
  16. - conv_b2: тензор формы (channel_2,) - смещения
  17. второго сверточного слоя.
  18. - fc_w: тензор представляющий весовые коэффициенты полносвязанного слоя.
  19. Могли бы определить сами, какая должна быть форма?
  20. - fc_b: тензор смещений полносвязанного слоя.
  21. Могли бы определить сами, какая должна быть форма?
  22. """
  23.  
  24. conv_w1, conv_b1, conv_w2, conv_b2, fc_w, fc_b = params
  25. scores = None
  26. ############################################################################
  27. # ЗАДАНИЕ: Реализуйте прямое распространение для 3-х слойной ConvNet. #
  28. ############################################################################
  29. # *****НАЧАЛО ВАШЕГО КОДА (НЕ УДАЛЯЙТЕ/НЕ МОДИФИЦИРУЙТЕ ЭТУ СТРОКУ)*****
  30.  
  31. x_padded = tf.pad(x, [[0,0], [2,2], [2,2], [0,0]], 'CONSTANT')
  32.  
  33. conv1 = tf.nn.conv2d(x_padded, conv_w1, [1, 1, 1, 1], padding='VALID') + conv_b1
  34. relu1 = tf.nn.relu(conv1)
  35.  
  36. conv1_padded = tf.pad(relu1, [[0,0], [1,1], [1,1], [0,0]],'CONSTANT')
  37.  
  38. conv2 = tf.nn.conv2d(conv1_padded, conv_w2, [1, 1, 1, 1], padding='VALID') + conv_b2
  39. relu2 = tf.nn.relu(conv2)
  40.  
  41. conv2_flattened = flatten(relu2)
  42.  
  43. fc1 = tf.matmul(conv2_flattened, fc_w) + fc_b
  44. scores = fc1
  45.  
  46. # *****КОНЕЦ ВАШЕГО КОДА (НЕ УДАЛЯЙТЕ/НЕ МОДИФИЦИРУЙТЕ ЭТУ СТРОКУ)*****
  47. ############################################################################
  48. # КОНЕЦ ВАШЕГО КОДА #
  49. ############################################################################
  50. return scores
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement