eryk.lewinson
/
mario_vs_wario_v2
connected to https://github.com/erykml/mario_vs_wario_v2.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
VIDEOS_DIR = "data/videos"
MODELS_DIR = "models"
RAW_IMAGES_DIR = "data/raw"
PROCESSED_IMAGES_DIR = "data/processed"

CLASSES = ["mario", "wario"]

YT_LINKS_DIR = {
    "mario": "https://www.youtube.com/watch?v=lXMJt5PP3kM",
    "wario": "https://www.youtube.com/watch?v=ZJuxdR0KH-s",
}

VIDEO_NAMES_DIR = {
    "mario": "Game Boy Longplay [005] Super Mario Land 2 6 Golden Coins.mp4",
    "wario": "Game Boy Longplay [004] Super Mario Land 3 Wario Land.mp4",
}

N_TRAIN = 2500
N_TEST = 500