summaryrefslogtreecommitdiff
path: root/run.py
diff options
context:
space:
mode:
Diffstat (limited to 'run.py')
-rw-r--r--run.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/run.py b/run.py
index 7e0fa46..cd19277 100644
--- a/run.py
+++ b/run.py
@@ -131,7 +131,7 @@ def dilate_frames(moduleNetwork, tensorOutput, frames, dilate):
firstFrame = nextFrame
nextFrame = frames[i]
new_frames += [firstFrame]
- new_frames += process_tree(module, firstFrame, nextFrame, tensorOutput, dilate, 1)
+ new_frames += process_tree(moduleNetwork, firstFrame, nextFrame, tensorOutput, dilate, 1)
new_frames += nextFrame
return new_frames