add installation of imaginaire

app.py

@@ -9,7 +9,9 @@ import test
 import importlib
 from scipy.interpolate import interp1d, splev, splprep
 import cv2
+import subprocess
 
+subprocess.run(["bash", "install_imaginaire.sh"])
 
 def get_single(sat_img, style_img, x_offset, y_offset):
     name = ''
@@ -139,21 +141,17 @@ def get_video(sat_img, style_img, positions):
 
     output_video_path = 'output_video.mp4'
 
-    # 设置视频的帧率、宽度和高度
     frame_rate = 15
     frame_width = 512
     frame_height = 128
 
-    # 使用OpenCV创建视频写入对象，选择H.264编码器
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(output_video_path, fourcc, frame_rate, (frame_width, frame_height))
 
-    # 遍历图像列表并将它们写入视频
     for image_np in rendered_image_list:
         image_np = cv2.cvtColor(image_np, cv2.COLOR_BGR2RGB)
         out.write(image_np)
 
-    # 释放视频写入对象
     out.release()
 
     return "output_video.mp4"

install_imaginaire.sh

@@ -0,0 +1,6 @@
+CURRENT=$(pwd)
+for p in correlation channelnorm resample2d bias_act upfirdn2d; do
+cd imaginaire/third_party/${p};
+python setup.py install;
+cd ${CURRENT};
+done