Add detection for kitchen recipe page

recipes.py

@@ -12,7 +12,7 @@
 # The expected color for the video background.
 BG_COLOR = (194, 222, 228)
 WOOD_COLOR = (115, 175, 228)
-KITCHEN_COLOR = (160, 167, 246)
+KITCHEN_COLOR = (160, 163, 243)
 
 # Items that look very alike and can be mixed up.
 CONFUSED_ITEMS = {
@@ -37,11 +37,11 @@ def __repr__(self):
 def detect(frame: numpy.ndarray) -> bool:
     """Detects if a given frame is showing DIY recipes."""
     color = frame[:20, 1200:1240].mean(axis=(0, 1))
-    if numpy.linalg.norm(color - WOOD_COLOR) < 7:
+    if numpy.linalg.norm(color - WOOD_COLOR) < 10:
         raise AssertionError('Workbench scanning is not supported.')
-    if numpy.linalg.norm(color - KITCHEN_COLOR) < 7:
+    if numpy.linalg.norm(color - KITCHEN_COLOR) < 10:
         raise AssertionError('Kitchen scanning is not supported.')
-    return numpy.linalg.norm(color - BG_COLOR) < 7
+    return numpy.linalg.norm(color - BG_COLOR) < 10
 
 
 def scan(video_file: str, locale: str = 'en-us') -> ScanResult:

scanner_test.py

@@ -102,8 +102,7 @@ def test_music_translate(self):
 
 class ScannerExtraTest(parameterized.TestCase):
 
-    @parameterized.named_parameters(
-        (k.split('.')[0], k) for k in EXTRA_GROUND_TRUTH)
+    @parameterized.parameters(EXTRA_GROUND_TRUTH)
     def test_extra(self, filename):
         filepath = os.path.join('examples/extra', filename)
         try:

testdata/extra.json

@@ -31,5 +31,6 @@
     "recipes_full.mp4": ["Aquarius urn","Aries rocking chair","Bunny Day arch","Bunny Day bag","Bunny Day bed","Bunny Day crown","Bunny Day fence","Bunny Day festive balloons","Bunny Day flooring","Bunny Day glowy garland","Bunny Day lamp","Bunny Day merry balloons","Bunny Day rug","Bunny Day stool","Bunny Day table","Bunny Day vanity","Bunny Day wall","Bunny Day wall clock","Bunny Day wand","Bunny Day wardrobe","Bunny Day wreath","Cancer table","Capricorn ornament","DIY workbench","Gemini closet","Jingle wall","King Tut mask","Leo sculpture","Libra scale","Pisces lamp","Roost sablé cookie","Sagittarius arrow","Scorpio lamp","Taurus bathtub","Turkey Day casserole","Turkey Day chair","Turkey Day decorations","Turkey Day garden stand","Turkey Day hearth","Turkey Day table","Turkey Day table setting","Turkey Day wheat decor","Virgo harp","acorn pochette","acorn rug","acoustic guitar","aji fry","anchoas al ajillo","angled signpost","apple chair","apple dress","apple hat","apple jam","apple jelly","apple pie","apple rug","apple smoothie","apple tart","apple umbrella","apple wall","armor shoes","aroma pot","asteroid","astronaut suit","autumn wall","axe","backyard lawn","baked potatoes","bamboo basket","bamboo bench","bamboo candleholder","bamboo doll","bamboo drum","bamboo floor lamp","bamboo flooring","bamboo hat","bamboo lattice fence","bamboo lunch box","bamboo noodle slide","bamboo partition","bamboo shelf","bamboo speaker","bamboo sphere","bamboo stool","bamboo stopblock","bamboo wall","bamboo wall decoration","bamboo wand","bamboo-grove wall","bamboo-shoot lamp","bamboo-shoot soup","bamboo-slats fence","barbed-wire fence","barbell","barred-knifejaw carpaccio","barrel","basement flooring","basket pack","beekeeper's hive","big festive tree","birdbath","birdcage","birdhouse","block fence","blossom-viewing lantern","blue rose crown","blue rose wreath","bone doorplate","bonfire","bonsai shelf","boomerang","box-shaped seat","bread","bread gratin","brick fence","brick oven","brick pillar","brick well","brown herringbone wall","brown sugar","brown-sugar cupcakes","brown-sugar pound cake","butter churn","cabin wall","cake salé","campfire","cardboard bed","cardboard chair","cardboard sofa","cardboard table","carp on a cutting board","carpaccio di capesante","carpaccio di marlin blu","carpaccio di salmone","carrot bagel sandwich","carrot cake","carrot juice","carrot potage","carrot scones","carrot-tops curry","cave","champiñones al ajillo","cherry dress","cherry hat","cherry jam","cherry jelly","cherry lamp","cherry pie","cherry rug","cherry smoothie","cherry speakers","cherry tart","cherry umbrella","cherry wall","cherry-blossom bonsai","cherry-blossom branches","cherry-blossom clock","cherry-blossom flooring","cherry-blossom pochette","cherry-blossom pond stone","cherry-blossom rug","cherry-blossom umbrella","cherry-blossom wand","cherry-blossom-petal pile","cherry-blossom-trees wall","chic cosmos wreath","chic mum crown","chic rose crown","chic tulip crown","chic windflower wreath","chocolate herringbone wall","clackercart","clam chowder","classic pitcher","classic-library wall","clothesline","coconut cookies","coconut juice","coconut milk","coconut oil","coconut pancakes","coconut pudding","coconut wall planter","colored-leaves flooring","concrete pillar","cookies","cool hyacinth crown","cool hyacinth wreath","cool pansy crown","cool pansy wreath","cool windflower crown","cool windflower wreath","corral fence","corrugated iron fence","cosmos crown","cosmos shower","cosmos wand","cosmos wreath","country fence","crescent-moon chair","crest doorplate","crewed spaceship","cute lily crown","cute rose crown","cutting board","dark bamboo bath mat","dark bamboo rug","dark cosmos crown","dark lily crown","dark lily wreath","dark rose wreath","dark tulip crown","dark tulip wreath","dark wooden-mosaic wall","decayed tree","decoy duck","deer decoration","deer scare","destinations signpost","dharma","document stack","doghouse","donation box","drinking fountain","earth-egg outfit","earth-egg shell","earth-egg shoes","egg party dress","egg party hat","falling-snow wall","fancy lily wreath","fancy mum wreath","fancy rose wreath","festival-lantern set","festive rug","festive top set","festive tree","festive wrapping paper","firewood","fish and chips","fish bait","fishing rod","flat garden rock","flimsy axe","flimsy fishing rod","flimsy net","flimsy shovel","flimsy watering can","floral swag","flour","flower stand","flying saucer","forbidden altar","forest flooring","forest wall","fossil doorplate","fountain","foxtail","french fries","frosted cookies","frosted pretzels","frozen arch","frozen bed","frozen chair","frozen counter","frozen fence","frozen floor tiles","frozen mini snowperson","frozen partition","frozen pillar","frozen sculpture","frozen table","frozen tree","frozen-treat set","fruit basket","fruit cupcakes","fruit pizza","fruit salad","fruit scones","fruit wreath","fruit-topped pancakes","frying pan","full-body glowing-moss suit","galaxy flooring","garbage-heap flooring","garbage-heap wall","garden bench","garden rock","garden wagon","gear apparatus","gear tower","giant ornament","giant teddy bear","giant vine","gift pile","glow-in-the-dark stickers","glowing-moss balloon","glowing-moss boulder","glowing-moss cave wall","glowing-moss dress","glowing-moss flooring","glowing-moss forest wall","glowing-moss headband","glowing-moss hood","glowing-moss jar","glowing-moss pointed cap","glowing-moss pond","glowing-moss rug","glowing-moss ruins wall","glowing-moss statue","glowing-moss stool","glowing-moss wreath","glowing-moss-jar shelves","gnocchi di carote","gnocchi di patate","gnocchi di zucca","gold armor","gold bars","gold helmet","gold rose crown","gold rose wreath","gold-armor shoes","gold-nugget mining car","gold-screen wall","golden altar","golden arowana model","golden axe","golden bathtub","golden candlestick","golden casket","golden decorative plate","golden dharma","golden dishes","golden dung beetle","golden flooring","golden garden bunny","golden gear apparatus","golden gear tower","golden gears","golden ladder set-up kit","golden meter and pipes","golden net","golden piggy bank","golden pillar","golden plate armor","golden rod","golden samurai suit","golden seat","golden shovel","golden slingshot","golden toilet","golden urn","golden vase","golden wall","golden wand","golden watering can","golden wristwatch","gong","grass skirt","grass standee","gratin","green bamboo fence","green bamboo mat","green grass skirt","green-leaf pile","grilled sea bass with herbs","hanging glowing moss","hanging terrarium","hay bed","hearth","hedge","hedge standee","holiday candle","honeycomb flooring","honeycomb wall","hyacinth crown","hyacinth lamp","hyacinth wand","hyacinth wreath","ice flooring","ice wall","ice wand","iceberg flooring","iceberg wall","illuminated present","illuminated reindeer","illuminated snowflakes","illuminated tree","imperial fence","imperial pot","infused-water dispenser","iron armor","iron closet","iron doorplate","iron fence","iron frame","iron garden bench","iron garden chair","iron garden table","iron hanger stand","iron ladder set-up kit","iron shelf","iron wall lamp","iron wall rack","iron wand","iron worktable","iron-and-stone fence","ironwood DIY workbench","ironwood bed","ironwood cart","ironwood chair","ironwood clock","ironwood cupboard","ironwood dresser","ironwood kitchenette","ironwood low table","ironwood table","jail bars","jarred bamboo shoots","jarred mushrooms","juicy-apple TV","jungle flooring","jungle wall","kabu ankake","karei no nitsuke","kettle bathtub","kettlebell","key holder","knight's helmet","knitted-grass backpack","ladder","large cardboard boxes","large lattice fence","lattice fence","leaf","leaf campfire","leaf mask","leaf stool","leaf umbrella","leaf-egg outfit","leaf-egg shell","leaf-egg shoes","light bamboo bath mat","light bamboo rug","lily crown","lily record player","lily wand","lily wreath","log bed","log bench","log chair","log decorative shelves","log dining table","log extra-long sofa","log garden lounge","log pack","log round table","log stakes","log stool","log wall-mounted clock","lovely cosmos crown","low brick island counter","low concrete island counter","low golden island counter","low marble island counter","low simple island counter","low steel island counter","low wooden island counter","lucky gold cat","lunar lander","lunar rover","lunar surface","magazine rack","manga-library wall","manhole cover","maple-leaf pochette","maple-leaf pond stone","maple-leaf rug","maple-leaf umbrella","marble pillar","matryoshka","medicine","medium cardboard boxes","medium wooden partition","mermaid bed","mermaid chair","mermaid closet","mermaid dresser","mermaid fence","mermaid flooring","mermaid lamp","mermaid rug","mermaid screen","mermaid shelf","mermaid sofa","mermaid table","mermaid vanity","mermaid wall","mermaid wall clock","minestrone soup","mini DIY workbench","mini golden dharma","mixed-fruits bagel sandwich","mixed-fruits crepe","mixed-fruits pie","mixed-fruits sandwich","mixed-fruits tart","modeling clay","modern wood wall","money flooring","moon","mossy garden rock","mountain standee","mum crown","mum cushion","mum wreath","mums wand","mush lamp","mush log","mush low stool","mush parasol","mush partition","mush table","mush umbrella","mush wall","mushroom crepe","mushroom curry","mushroom pizza","mushroom potage","mushroom salad","mushroom wand","mushroom wreath","music stand","natural garden chair","natural garden table","natural mum wreath","natural square table","net","nice branch","nova light","ocarina","oil-barrel bathtub","old-fashioned washtub","olive-flounder meunière","open wooden shelves","orange dress","orange end table","orange hat","orange jelly","orange marmalade","orange pie","orange pound cake","orange rug","orange smoothie","orange tart","orange umbrella","orange wall","orange wall-mounted clock","organic bread","ornament crown","ornament garland","ornament mobile","ornament table lamp","ornament tree","ornament wreath","outdoor bath","outdoor picnic set","palm-tree lamp","pan flute","pancakes","pansy crown","pansy table","pansy wand","pansy wreath","park fence","partition wall","paw-print doorplate","peach chair","peach dress","peach hat","peach jam","peach jelly","peach pie","peach rug","peach smoothie","peach surprise box","peach tart","peach umbrella","peach wall","pear bed","pear dress","pear hat","pear jam","pear jelly","pear pie","pear rug","pear smoothie","pear tart","pear umbrella","pear wall","pear wardrobe","pesce all'acqua pazza","pickled veggies","pile of cardboard boxes","pile of cash","pile of leaves","pile of zen cushions","pine bonsai tree","pitfall seed","pizza margherita","plain cupcakes","plain scones","plain sink","plain wooden shop sign","plate armor","poke","pond stone","pot","potato curry","potato galette","potato potage","potted ivy","pound cake","pretty cosmos wreath","pretty tulip wreath","pretzels","project table","pull-apart bread","pumpkin bagel sandwich","pumpkin cupcakes","pumpkin curry","pumpkin pie","pumpkin pound cake","pumpkin scones","pumpkin soup","purple hyacinth crown","purple hyacinth wreath","purple pansy crown","purple windflower crown","raccoon figurine","rainbow feather","recycled boots","recycled-can thumb piano","red-leaf pile","ringtoss","robot hero","rocket","rocking chair","rocking horse","rope fence","rope-net flooring","rope-net wall","rose bed","rose crown","rose wand","rose wreath","round glowing-moss rug","round vine rug","ruined arch","ruined broken pillar","ruined decorated pillar","ruined seat","rustic-stone wall","sakura-wood flooring","sakura-wood wall","salad","salad-stuffed tomato","salade de carottes râpées","salmon bagel sandwich","salmon sandwich","sandy-beach flooring","sardines in oil","satellite","sauna heater","sautéed olive flounder","savory bread","scarecrow","scattered papers","sci-fi flooring","sci-fi wall","sea-bass pie","seafood ajillo","seafood pizza","seafood salad","seaweed soup","senmaizuke barrel","shamrock wand","shell arch","shell bed","shell fountain","shell lamp","shell partition","shell rug","shell speaker","shell stool","shell table","shell wand","shell wreath","shellfish pochette","shovel","signpost","silo","simple DIY workbench","simple mum crown","simple pillar","simple well","simple wooden fence","skateboard wall rack","ski-slope flooring","ski-slope wall","sky-egg outfit","sky-egg shell","sky-egg shoes","sleigh","slingshot","small cardboard boxes","small wooden partition","snack bread","snazzy pansy wreath","snowflake pochette","snowflake wall","snowflake wreath","snowperson head","space shuttle","spaghetti marinara","spaghetti napolitan","spiky fence","spooky arch","spooky candy set","spooky carriage","spooky chair","spooky cookies","spooky fence","spooky garland","spooky lantern","spooky lantern set","spooky scarecrow","spooky standing lamp","spooky table","spooky table setting","spooky tower","spooky treats basket","spooky tree","spooky trick lamp","spooky wand","squid-ink curry","squid-ink spaghetti","stack of books","stacked bottle crates","stacked fish containers","stacked magazines","stacked senmaizuke barrels","stacked shopping baskets","stacked-wood wall","stall","standard umbrella stand","star clock","star head","star pochette","star wand","starry garland","starry wall","starry-sands flooring","starry-skies rug","starry-sky wall","steamer-basket set","steel flooring","steel pillar","steel-frame wall","stone arch","stone axe","stone fence","stone lion-dog","stone stool","stone table","stone tablet","stone wall","stone-egg outfit","stone-egg shell","stone-egg shoes","stonework kitchen","straw fence","straw umbrella hat","street piano","succulent plant","sugar","sugar crepe","summer-shell rug","suspicious cauldron","swinging bench","tabletop festive tree","tall brick island counter","tall concrete island counter","tall garden rock","tall golden island counter","tall lantern","tall marble island counter","tall simple island counter","tall steel island counter","tall wooden island counter","tea table","tension-pole rack","terrarium","three-tiered snowperson","thumbprint jam cookies","tiki torch","timber doorplate","tiny library","tire stack","tire toy","tomates al ajillo","tomato bagel sandwich","tomato curry","tomato juice","tomato puree","traditional balancing toy","traditional straw coat","trash bags","tree branch wreath","tree standee","tree's bounty arch","tree's bounty big tree","tree's bounty lamp","tree's bounty little tree","tree's bounty mobile","tree-branch wand","tree-stump rug","trophy case","tropical vista","tulip crown","tulip surprise box","tulip wand","tulip wreath","turnip salad","ukulele","underwater flooring","underwater wall","unglazed dish set","vaulting pole","veggie cookies","veggie crepe","veggie cupcakes","veggie quiche","veggie sandwich","vertical-board fence","vine bench","vine crown","vine garland","vine hanging chair","vine ladder set-up kit","vine lamp","vine outfit","wand","water flooring","water pump","water-egg outfit","water-egg shell","water-egg shoes","watering can","wave breaker","wedding fence","wedding wand","western-style stone","whole-wheat flour","wild log bench","wild-wood wall","windflower crown","windflower fan","windflower wand","windflower wreath","wobbling Zipper toy","wood-egg outfit","wood-egg shell","wood-egg shoes","wooden bookshelf","wooden box","wooden bucket","wooden chair","wooden chest","wooden double bed","wooden end table","wooden field sign","wooden fish","wooden full-length mirror","wooden ladder set-up kit","wooden low table","wooden mini table","wooden music box","wooden pillar","wooden simple bed","wooden stool","wooden storage shed","wooden table","wooden table mirror","wooden toolbox","wooden wardrobe","wooden waste bin","wooden-block bed","wooden-block bench","wooden-block bookshelf","wooden-block chair","wooden-block chest","wooden-block stereo","wooden-block stool","wooden-block table","wooden-block toy","wooden-block wall clock","wooden-knot wall","wooden-mosaic wall","wooden-plank sign","woodland wall","woven-vine pochette","woven-vines hat","yellow bamboo mat","yellow star rug","yellow-leaf pile","zen fence","zen-style stone"],
     "recipes_img.jpg": ["Bunny Day wand", "cosmos wand", "lily wand", "mums wand", "spooky wand", "wedding wand", "windflower wand"],
     "recipes_wood.mp4": "Workbench scanning is not supported.",
-    "recipes_kitchen.jpg": "Kitchen scanning is not supported."
+    "recipes_kitchen.jpg": "Kitchen scanning is not supported.",
+    "recipes_kitchen.mp4": "Kitchen scanning is not supported."
 }