Selaa lähdekoodia

feat: add JSONProducer

Outputs a representation of the Features extrated by the pipeline

Intent is to write a FE that takes the output so that a pipeline can be
're-run'

Output JSON could also be used with external tools
main
Rob Hallam 2 kuukautta sitten
vanhempi
commit
4ecc95174e
1 muutettua tiedostoa jossa 13 lisäystä ja 0 poistoa
  1. +13
    -0
      pipeline/video_producers.py

+ 13
- 0
pipeline/video_producers.py Näytä tiedosto

@@ -1,6 +1,7 @@
"""Classes for producing videos"""

from abc import ABC
import json
import logging
import os
import subprocess
@@ -139,3 +140,15 @@ class VisualisationProducer(Producer):
# save the plot
plt.savefig("/tmp/visualisation.png")
plt.close()

class JSONProducer(Producer):
"""Produce JSON output"""
def __init__(self, features):
if not features:
raise ValueError("No features provided")
self.features = features

def produce(self):
# FIXME: config option for output path
with open("/tmp/features.json", "w") as jsonfile:
jsonfile.write(json.dumps(self.features, cls=PipelineJSONEncoder, indent=4))

Ladataan…
Peruuta
Tallenna