Victoria Oberascher commited on
Commit
3eb0c8a
·
1 Parent(s): 0c1d03a

update readme

Browse files
Files changed (1) hide show
  1. README.md +5 -4
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
- # Use artificial data for testing or
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
- # Load data from fiftyone
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
- # Get the ground truth points
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
- # Get the predicted points
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 = [