gseps_inference/config/inference_config.json

28 lines
974 B
JSON

{
"slack_url": "https://hooks.slack.com/services/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",
"slack_message": "Inference Done.",
"amqp_url": "localhost",
"amqp_port": 5672,
"amqp_vhost": "/",
"amqp_id": "worker",
"amqp_pw": "gseps1234",
"amqp_taskq": "TaskQ",
"amqp_resultq": "ResultQ",
"model_config": {
"points_per_side": 36,
"pred_iou_thresh": 0.86,
"stability_score_thresh": 0.9,
"crop_n_layers": 1,
"crop_n_points_downscale_factor": 1,
"box_nms_thresh": 0.8,
"min_mask_region_area": 10,
"area_thresh": 0.1,
"device": "cuda"
},
"model_checkpoints": "/home/sdt/Workspace/gseps/rabbitmq_test/weights/sam_vit_h_4b8939.pth",
"remote_server_ip": "25.15.14.31",
"remote_server_id": "sdt",
"remote_server_pw": "251327",
"copied_image_path_from_remote_server": "/home/sdt/Workspace/gseps/rabbitmq_test/image_bucket",
"inference_result_path": "/home/sdt/Workspace/gseps/rabbitmq_test/inference_result/"
}