Spaces:
Build error
Build error
Victoria Oberascher
commited on
Commit
·
3eb0c8a
1
Parent(s):
0c1d03a
update readme
Browse files
README.md
CHANGED
@@ -35,9 +35,10 @@ This is how you can quickly evaluate your horizon prediction models using SEA-AI
|
|
35 |
|
36 |
```python
|
37 |
|
|
|
38 |
import evaluate
|
39 |
|
40 |
-
#
|
41 |
ground_truth_points = [[[0.0, 0.5384765625], [1.0, 0.4931640625]],
|
42 |
[[0.0, 0.53796875], [1.0, 0.4928515625]],
|
43 |
[[0.0, 0.5374609375], [1.0, 0.4925390625]],
|
@@ -50,20 +51,20 @@ prediction_points = [[[0.0, 0.5428930956049597], [1.0, 0.4642497615378973]],
|
|
50 |
[[0.0, 0.5200016849393765], [1.0, 0.4728554579177664]],
|
51 |
[[0.0, 0.523573113510805], [1.0, 0.47642688648919496]]]
|
52 |
|
53 |
-
#
|
54 |
sequence = "Sentry_2023_02_Portugal_2023_01_24_19_15_17"
|
55 |
dataset_name = "SENTRY_VIDEOS_DATASET_QA"
|
56 |
sequence_view = fo.load_dataset(dataset_name).match(F("sequence") == sequence)
|
57 |
sequence_view = sequence_view.select_group_slices("thermal_wide")
|
58 |
|
59 |
-
#
|
60 |
polylines_gt = sequence_view.values("frames.ground_truth_pl")
|
61 |
ground_truth_points = [
|
62 |
line["polylines"][0]["points"][0] for line in polylines_gt[0]
|
63 |
if line is not None
|
64 |
]
|
65 |
|
66 |
-
#
|
67 |
polylines_pred = sequence_view.values(
|
68 |
"frames.ahoy-IR-b2-whales__XAVIER-AGX-JP46_pl")
|
69 |
prediction_points = [
|
|
|
35 |
|
36 |
```python
|
37 |
|
38 |
+
|
39 |
import evaluate
|
40 |
|
41 |
+
#Use artificial data for testing or
|
42 |
ground_truth_points = [[[0.0, 0.5384765625], [1.0, 0.4931640625]],
|
43 |
[[0.0, 0.53796875], [1.0, 0.4928515625]],
|
44 |
[[0.0, 0.5374609375], [1.0, 0.4925390625]],
|
|
|
51 |
[[0.0, 0.5200016849393765], [1.0, 0.4728554579177664]],
|
52 |
[[0.0, 0.523573113510805], [1.0, 0.47642688648919496]]]
|
53 |
|
54 |
+
#Load data from fiftyone
|
55 |
sequence = "Sentry_2023_02_Portugal_2023_01_24_19_15_17"
|
56 |
dataset_name = "SENTRY_VIDEOS_DATASET_QA"
|
57 |
sequence_view = fo.load_dataset(dataset_name).match(F("sequence") == sequence)
|
58 |
sequence_view = sequence_view.select_group_slices("thermal_wide")
|
59 |
|
60 |
+
#Get the ground truth points
|
61 |
polylines_gt = sequence_view.values("frames.ground_truth_pl")
|
62 |
ground_truth_points = [
|
63 |
line["polylines"][0]["points"][0] for line in polylines_gt[0]
|
64 |
if line is not None
|
65 |
]
|
66 |
|
67 |
+
#Get the predicted points
|
68 |
polylines_pred = sequence_view.values(
|
69 |
"frames.ahoy-IR-b2-whales__XAVIER-AGX-JP46_pl")
|
70 |
prediction_points = [
|