| { | |
| "yolo": { | |
| "precision": 0.9849494446224883, | |
| "recall": 0.9915590449234907, | |
| "map50": 0.9947342360177375, | |
| "map50_95": 0.8924805724717391 | |
| }, | |
| "rl": { | |
| "capture_rate": 0.16666666666666666, | |
| "avg_reward": 341.5984839566094, | |
| "avg_min_dist": 0.20183070997397104, | |
| "avg_ep_length": 168.03333333333333, | |
| "n_eval": 30 | |
| }, | |
| "config": { | |
| "IMG_SIZE": 256, | |
| "YOLO_MODEL": "yolov8n.pt", | |
| "YOLO_EPOCHS": 15, | |
| "YOLO_BATCH": 16, | |
| "MAX_TRAIN": 4000, | |
| "MAX_VAL": 600, | |
| "MAX_SAT": 500, | |
| "RL_TIMESTEPS": 100000, | |
| "RL_N_ENVS": 4, | |
| "RL_MAX_STEPS": 200, | |
| "EVAL_EPISODES": 30, | |
| "CAPTURE_THRESH": 0.18, | |
| "DELTA_SCALE": 0.05, | |
| "IK_SUBSTEPS": 5, | |
| "SEED": 42, | |
| "DEVICE": "cuda" | |
| } | |
| } |