reconocimiento-facial / config.json
CarPeAs's picture
Commit inicial del modelo de reconocimiento facial de expresiones
c5edfaa verified
raw
history blame contribute delete
470 Bytes
{"model_name": "resnet50", "num_labels": 7, "image_size": 224, "pretrained": true, "input_channels": 3, "output_activation": "softmax", "loss_function": "CrossEntropyLoss", "optimizer": "Adam", "learning_rate": 0.001, "transformations": {"Resize": [224, 224], "Grayscale": false, "RandomHorizontalFlip": true, "ColorJitter": {"brightness": 0.2, "contrast": 0.2, "saturation": 0.2, "hue": 0.1}, "Normalize": {"mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225]}}}