Вычисление стоимости резьбы шва за один цикл в питоне

Question

Как вычислить стоимость энергии для вырезания шва в одном цикле, итерируя по строкам для python? Вики вырезания шва Как динамическое программирование в вики, мне нужно min_cost в последнем ряду для возможных трех ячеек, а также хранить стоимость и путь.

И это очень медленно, используя два цикла, так что кто-нибудь знает, как сделать это более эффективно?

Answer 1

Вы можете использовать numba.jit, чтобы (возможно) ускорить вычисления, при условии, что вы соблюдаете правильную типизацию. Невозможно избежать 2-х циклов в динамическом программировании, однако вы можете взглянуть на улучшенную резьбу швов (которая также дает лучшие результаты в целом)

https://github.com/axu2/improved-seam-carvinghttps://medium.com/@avik.das/improved-seam-carving-with-forward-energy-88ba84dab7e

      from numba import jit 

@jit
def calc_seam(img):
   ...