gpu

app.py

@@ -87,7 +87,7 @@ for model_info in models_info:
 
 x = load_data()
 test_imgs, points = load_keypoints()
-mean_vector_list = []
+mean_vectors_list = []
 model_index = 0
 
 # ヒートマップの生成関数
@@ -149,6 +149,7 @@ def setup(model_info, input_image=None):
 
     feature_map, _ = models[model_index](test_imgs)
     mean_vector_list = utils.get_mean_vector(feature_map, points)
+    print("mean_vector_list", mean_vector_list)
     
     if input_image is not None:
         fig = get_heatmaps(0, image_size // 2, image_size // 2, input_image)