Spaces:
Runtime error
Runtime error
hecl
commited on
Commit
·
d6a239d
1
Parent(s):
0da2d4c
- [MINOR] [SOURCE] [UPDATE] 1. update app.py
Browse files
app.py
CHANGED
|
@@ -73,44 +73,40 @@ def compare_score(score, score_list):
|
|
| 73 |
def create_bar_chart(scores, comparisons):
|
| 74 |
labels = ['Semantic', 'Aesthetic', 'Technical', 'Overall']
|
| 75 |
base_colors = ['#d62728', '#ff7f0e', '#1f77b4', '#bcbd22']
|
| 76 |
-
|
| 77 |
fig, ax = plt.subplots(figsize=(8, 6))
|
| 78 |
|
| 79 |
-
# Create
|
| 80 |
bars = ax.bar(labels, scores, color=base_colors, edgecolor='black', width=0.6)
|
| 81 |
|
| 82 |
-
# Add gradient effect
|
| 83 |
-
for bar, color in zip(bars, base_colors):
|
| 84 |
-
bar.set_facecolor(plt.cm.get_cmap('autumn')(bar.get_height() / 5))
|
| 85 |
-
|
| 86 |
# Adding the text labels for scores
|
| 87 |
for bar, score in zip(bars, scores):
|
| 88 |
height = bar.get_height()
|
| 89 |
-
ax.annotate(f'{score}',
|
| 90 |
xy=(bar.get_x() + bar.get_width() / 2, height),
|
| 91 |
xytext=(0, 3), # 3 points vertical offset
|
| 92 |
textcoords="offset points",
|
| 93 |
ha='center', va='bottom',
|
| 94 |
color='black')
|
| 95 |
-
|
| 96 |
# Add comparison text
|
| 97 |
comparison_texts = []
|
| 98 |
-
for i, score in enumerate(scores):
|
| 99 |
percentage = (np.sum(comparisons[i] < score) / len(comparisons[i])) * 100
|
| 100 |
comparison_text = f'Better than {percentage:.0f}% videos in YT-UGC' if percentage > 50 else f'Worse than {100 - percentage:.0f}% videos in YT-UGC'
|
| 101 |
comparison_texts.append(comparison_text)
|
| 102 |
ax.annotate(comparison_text,
|
| 103 |
-
xy=(bar.get_x() + bar.get_width(),
|
| 104 |
xytext=(5, 0), # 5 points horizontal offset
|
| 105 |
textcoords="offset points",
|
| 106 |
ha='left', va='center',
|
| 107 |
-
color=
|
| 108 |
-
|
| 109 |
ax.set_xlabel('Categories')
|
| 110 |
ax.set_ylabel('Scores')
|
| 111 |
ax.set_ylim(0, 5)
|
| 112 |
ax.set_title('Video Quality Scores')
|
| 113 |
-
|
| 114 |
plt.tight_layout()
|
| 115 |
image_path = "./scores_bar_chart.png"
|
| 116 |
plt.savefig(image_path)
|
|
|
|
| 73 |
def create_bar_chart(scores, comparisons):
|
| 74 |
labels = ['Semantic', 'Aesthetic', 'Technical', 'Overall']
|
| 75 |
base_colors = ['#d62728', '#ff7f0e', '#1f77b4', '#bcbd22']
|
| 76 |
+
|
| 77 |
fig, ax = plt.subplots(figsize=(8, 6))
|
| 78 |
|
| 79 |
+
# Create vertical bars
|
| 80 |
bars = ax.bar(labels, scores, color=base_colors, edgecolor='black', width=0.6)
|
| 81 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 82 |
# Adding the text labels for scores
|
| 83 |
for bar, score in zip(bars, scores):
|
| 84 |
height = bar.get_height()
|
| 85 |
+
ax.annotate(f'{score:.1f}',
|
| 86 |
xy=(bar.get_x() + bar.get_width() / 2, height),
|
| 87 |
xytext=(0, 3), # 3 points vertical offset
|
| 88 |
textcoords="offset points",
|
| 89 |
ha='center', va='bottom',
|
| 90 |
color='black')
|
| 91 |
+
|
| 92 |
# Add comparison text
|
| 93 |
comparison_texts = []
|
| 94 |
+
for i, (bar, score) in enumerate(zip(bars, scores)):
|
| 95 |
percentage = (np.sum(comparisons[i] < score) / len(comparisons[i])) * 100
|
| 96 |
comparison_text = f'Better than {percentage:.0f}% videos in YT-UGC' if percentage > 50 else f'Worse than {100 - percentage:.0f}% videos in YT-UGC'
|
| 97 |
comparison_texts.append(comparison_text)
|
| 98 |
ax.annotate(comparison_text,
|
| 99 |
+
xy=(bar.get_x() + bar.get_width(), bar.get_height() / 2),
|
| 100 |
xytext=(5, 0), # 5 points horizontal offset
|
| 101 |
textcoords="offset points",
|
| 102 |
ha='left', va='center',
|
| 103 |
+
color=base_colors[i])
|
| 104 |
+
|
| 105 |
ax.set_xlabel('Categories')
|
| 106 |
ax.set_ylabel('Scores')
|
| 107 |
ax.set_ylim(0, 5)
|
| 108 |
ax.set_title('Video Quality Scores')
|
| 109 |
+
|
| 110 |
plt.tight_layout()
|
| 111 |
image_path = "./scores_bar_chart.png"
|
| 112 |
plt.savefig(image_path)
|