File size: 339 Bytes
db24a4e
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from frame import Frame 

class Candidate:
    def __init__(self, frame: Frame, score: float):
        self.frame = frame
        self.score = score

    def serialize(self) -> dict:
        return {
            'video': self.frame.video,
            'frame_name': self.frame.frame_name + '.jpg',
            'score': self.score,
        }