fix ZeroDivisionError when multi gpu

vbench/aesthetic_quality.py

@@ -74,7 +74,8 @@ def laion_aesthetic(aesthetic_model, clip_model, video_list, device):
         num += 1
         video_results.append({'video_path': video_path, 'video_results': cur_avg.item()})
 
-    aesthetic_avg /= num
+    if num != 0:
+        aesthetic_avg /= num
     return aesthetic_avg, video_results
 
 

vbench/appearance_style.py

@@ -70,7 +70,7 @@ def appearance_style(clip_model, video_dict, device, sample="rand"):
                     'video_results': video_sim, 
                     'frame_results': cur_video,
                     'cur_sim': cur_sim})
-    sim_per_frame = sim / cnt
+    sim_per_frame = sim / cnt if cnt != 0 else None
     return sim_per_frame, video_results
 
 def compute_appearance_style(json_dir, device, submodules_list, **kwargs):

vbench/background_consistency.py

@@ -61,7 +61,7 @@ def background_consistency(clip_model, preprocess, video_list, device, read_fram
             'video_sim': video_sim,
             'cnt_per_video': cnt_per_video})
     # sim_per_video = sim / (len(video_list) - 1)
-    sim_per_frame = sim / cnt
+    sim_per_frame = sim / cnt if cnt != 0 else None
     return sim_per_frame, video_results
 
 

vbench/color.py

@@ -86,7 +86,7 @@ def color(model, video_dict, device):
                     'video_path': video_path, 
                     'video_results': cur_success_frame_rate,
                     'cur_success_frame_rate': cur_success_frame_rate,})
-    success_rate = success_frame_count_all / video_count
+    success_rate = success_frame_count_all / video_count if video_count != 0 else None
     return success_rate, video_results
 
 

vbench/human_action.py

@@ -106,7 +106,7 @@ def human_action(umt_path, video_list, device):
             'video_results': flag,
             'cor_num_per_video': cor_num_per_video,})
     # print(f"cor num: {cor_num}, total: {cnt}")
-    acc = cor_num / cnt
+    acc = cor_num / cnt if cnt != 0 else None
     return acc, video_results
 
 

vbench/imaging_quality.py

@@ -51,7 +51,7 @@ def technical_quality(model, video_list, device, **kwargs):
             score = model(frame)
             acc_score_video += float(score)
         video_results.append({'video_path': video_path, 'video_results': acc_score_video/len(images)})
-    average_score = sum([o['video_results'] for o in video_results]) / len(video_results)
+    average_score = sum([o['video_results'] for o in video_results]) / len(video_results) if video_results else 0
     average_score = average_score / 100.
     return average_score, video_results
 

vbench/multiple_objects.py

@@ -69,7 +69,7 @@ def multiple_objects(model, video_dict, device):
                 'video_results': cur_success_frame_rate,
                 'success_frame_count': cur_success_frame_count,
                 'frame_count': len(cur_video_pred)})
-    success_rate = success_frame_count / frame_count
+    success_rate = success_frame_count / frame_count if frame_count != 0 else None
     return success_rate, video_results
 
 

vbench/object_class.py

@@ -65,7 +65,7 @@ def object_class(model, video_dict, device):
                 'video_results': cur_success_frame_rate,
                 'success_frame_count': cur_success_frame_count,
                 'frame_count': len(cur_video_pred)})
-    success_rate = success_frame_count / frame_count
+    success_rate = success_frame_count / frame_count if frame_count != 0 else None
     return success_rate, video_results
 
 

vbench/scene.py

@@ -59,7 +59,7 @@ def scene(model, video_dict, device):
                 'video_results': cur_success_frame_rate,
                 'success_frame_count': cur_success_frame_count,
                 'frame_count': len(cur_video_pred)})
-    success_rate = success_frame_count / frame_count
+    success_rate = success_frame_count / frame_count if frame_count != 0 else None
     return success_rate, video_results
 
 

vbench/subject_consistency.py

@@ -65,7 +65,7 @@ def subject_consistency(model, video_list, device, read_frame):
         sim += video_sim
         video_results.append({'video_path': video_path, 'video_results': sim_per_images})
     # sim_per_video = sim / (len(video_list) - 1)
-    sim_per_frame = sim / cnt
+    sim_per_frame = sim / cnt if cnt != 0 else None
     return sim_per_frame, video_results