renamed AI_Model to AIModel for consistency, added it as a functioning drop-down select on settings page, added face_distance to db and code, put face_distance model_used into all classes ready for use

This commit is contained in:
2021-07-25 15:13:39 +10:00
parent f6f67b8a69
commit 555ce70577
5 changed files with 42 additions and 17 deletions

View File

@@ -227,6 +227,14 @@ class Refimg(Base):
def __repr__(self):
return f"<id: {self.id}, fname: {self.fname}, created_on: {self.created_on}>"
class AIModel(Base):
__tablename__ = "ai_model"
id = Column(Integer, primary_key=True )
name = Column(String)
def __repr__(self):
return f"<id: {self.id}, name: {self.name}>"
class Face(Base):
__tablename__ = "face"
id = Column(Integer, Sequence('face_id_seq'), primary_key=True )
@@ -240,6 +248,7 @@ class FaceFileLink(Base):
__tablename__ = "face_file_link"
face_id = Column(Integer, ForeignKey("face.id"), primary_key=True )
file_eid = Column(Integer, ForeignKey("file.eid"), primary_key=True )
model_used = Column(Integer, ForeignKey("ai_model.id") )
def __repr__(self):
return f"<face_id: {self.face_id}, file_eid={self.file_eid}"
@@ -248,6 +257,8 @@ class FaceRefimgLink(Base):
__tablename__ = "face_refimg_link"
face_id = Column(Integer, ForeignKey("face.id"), primary_key=True )
refimg_id = Column(Integer, ForeignKey("refimg.id"), primary_key=True )
model_used = Column(Integer, ForeignKey("ai_model.id") )
face_distance = Column(Integer)
def __repr__(self):
return f"<face_id: {self.face_id}, refimg_id={self.refimg_id}"