Skip to content

Commit

Permalink
Merge pull request #107 from clamsproject/102-fix-label-overpopulation
Browse files Browse the repository at this point in the history
removes slate subtypes from training label set
  • Loading branch information
keighrim authored Jun 26, 2024
2 parents dede3ef + a3df90e commit 9a000e8
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
4 changes: 2 additions & 2 deletions metadata.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,11 +30,11 @@ def appmetadata() -> AppMetadata:
# which had default mappings for each model.
labelMap = [
"B:bars",
"S:slate", "S-H:slate", "S-C:slate", "S-D:slate", "S-G:slate",
"S:slate",
"W:other_opening", "L:other_opening", "O:other_opening", "M:other_opening",
"I:chyron", "N:chyron", "Y:chyron",
"C:credit", "R:credit",
"E:other_text", "K:other_text", "G:other_text", "T:other_text", "F:other_text" ]
"E:other_text", "K:other_text", "G:other_text", "T:other_text", "F:other_text"]

metadata = AppMetadata(
name="Scenes-with-text Detection",
Expand Down
4 changes: 3 additions & 1 deletion modeling/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,10 @@
positive_label = 'POS'

# full typology from https://github.com/clamsproject/app-swt-detection/issues/1
FRAME_TYPES = ["B", "S", "S:H", "S:C", "S:D", "S:B", "S:G", "W", "L", "O",
FRAME_TYPES = ["B", "S", "W", "L", "O",
"M", "I", "N", "E", "P", "Y", "K", "G", "T", "F", "C", "R"]
FRAME_TYPES_WITH_SUBTYPES = ["B", "SH", "SC", "SD", "SB", "SG", "W", "L", "O",
"M", "I", "N", "E", "P", "Y", "K", "G", "T", "F", "C", "R"]

# These are time frames that are typically static (that is, the text does not
# move around or change as with rolling credits). These are frame names after
Expand Down

0 comments on commit 9a000e8

Please sign in to comment.