zenon/master-thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Move export function to helper module

Browse Source
This commit is contained in:
Tobias Eidelpes 2023-02-28 17:45:51 +01:00
parent 2cfe51d478
commit 26353903af
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
classification/evaluation/helpers.py
View File

@ -2,6 +2,7 @@ import matplotlib.pyplot as plt
import pandas as pd import pandas as pd
import numpy as np import numpy as np
import seaborn as sns import seaborn as sns
import fiftyone as fo
def set_size(width, fraction=1, subplots=(1, 1)): def set_size(width, fraction=1, subplots=(1, 1)):
@ -37,3 +38,21 @@ def set_size(width, fraction=1, subplots=(1, 1)):
fig_dim = (fig_width_in, fig_height_in) fig_dim = (fig_width_in, fig_height_in)
return fig_dim return fig_dim
def export_dataset(dataset, export_dir, classes=["Healthy", "Stressed"]):
label_field = "ground_truth"
# The splits to export
splits = ["val"]
# Export the splits
for split in splits:
split_view = dataset.match_tags(split)
split_view.export(
export_dir=export_dir,
dataset_type=fo.types.YOLOv5Dataset,
label_field=label_field,
split=split,
classes=classes,
)