| { | |
| "class_names": [ | |
| "PG", | |
| "PG13", | |
| "R", | |
| "X", | |
| "XXX" | |
| ], | |
| "id2label": { | |
| "0": "PG", | |
| "1": "PG13", | |
| "2": "R", | |
| "3": "X", | |
| "4": "XXX" | |
| }, | |
| "image_models": [ | |
| "baseresNet18", | |
| "resNet50CV", | |
| "vitRater" | |
| ], | |
| "label2id": { | |
| "PG": 0, | |
| "PG13": 1, | |
| "R": 2, | |
| "X": 3, | |
| "XXX": 4 | |
| }, | |
| "model_type": "multimodal", | |
| "models": { | |
| "baseresNet18": "./models/baseresNet18", | |
| "promptBert": "./models/promptBert", | |
| "promptRoberta": "./models/promptRoberta", | |
| "resNet50CV": "./models/resNet50CV", | |
| "vitRater": "./models/vitRater" | |
| }, | |
| "nlp_transformers_model_paths": { | |
| "promptBert": "./models/promptBert", | |
| "promptRoberta": "./models/promptRoberta" | |
| }, | |
| "resnet_model_paths": { | |
| "baseresNet18": "./models/baseresNet18", | |
| "resNet50CV": "./models/resNet50CV" | |
| }, | |
| "text_models": [ | |
| "promptBert", | |
| "promptRoberta" | |
| ], | |
| "transformers_version": "4.42.4", | |
| "vit_model_paths": { | |
| "vitRater": "./models/vitRater" | |
| }, | |
| "yolo_model_paths": {} | |
| } | |