diff --git a/config/ftp_config.json b/config/ftp_config.json index af8c473..bbff9ac 100644 --- a/config/ftp_config.json +++ b/config/ftp_config.json @@ -3,5 +3,5 @@ "ftp_port": 21, "ftp_id": "sdt", "ftp_pw": "251327", - "ftp_root_dir": "/home/sdt/Workspace/gseps/rabbitmq_test/inference_result/" -} \ No newline at end of file + "ftp_root_dir": "/home/sdt/Workspace/gseps_inference/inference_result/" +} diff --git a/config/inference_config.json b/config/inference_config.json index 2c0a7b1..a0e0374 100644 --- a/config/inference_config.json +++ b/config/inference_config.json @@ -19,10 +19,10 @@ "area_thresh": 0.1, "device": "cuda" }, - "model_checkpoints": "/home/sdt/Workspace/gseps/rabbitmq_test/weights/sam_vit_h_4b8939.pth", + "model_checkpoints": "/home/sdt/Workspace/gseps_inference/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/" -} \ No newline at end of file + "copied_image_path_from_remote_server": "/home/sdt/Workspace/gseps_inference/image_bucket", + "inference_result_path": "/home/sdt/Workspace/gseps_inference/inference_result/" +} diff --git a/ftp_server.py b/ftp_server.py index 548e21d..d33c4f0 100644 --- a/ftp_server.py +++ b/ftp_server.py @@ -4,7 +4,7 @@ from pyftpdlib.handlers import FTPHandler from pyftpdlib.servers import FTPServer # config -with open('./config/ftp_config.json', 'r') as f: +with open('/home/sdt/Workspace/gseps_inference/config/ftp_config.json', 'r') as f: info = json.load(f) # 서버 설정 diff --git a/ftp_server.service b/ftp_server.service index 9c37bac..217b0f7 100644 --- a/ftp_server.service +++ b/ftp_server.service @@ -4,6 +4,8 @@ Description=FTP Server [Service] ExecStart=/home/sdt/miniconda3/bin/python /home/sdt/Workspace/gseps/ftp_server.py Restart=on-failure +Group=sdt +User=sdt [Install] WantedBy=multi-user.target diff --git a/inference.py b/inference.py index ba2590b..2a13fd8 100644 --- a/inference.py +++ b/inference.py @@ -18,7 +18,7 @@ from segment_anything import sam_model_registry, SamAutomaticMaskGenerator ############################################### # Config # ############################################### -with open('./config/inference_config.json', 'r') as f: +with open('/home/sdt/Workspace/gseps_inference/config/inference_config.json', 'r') as f: info = json.load(f) diff --git a/inference_processor.service b/inference_processor.service index 25f60af..dad465b 100644 --- a/inference_processor.service +++ b/inference_processor.service @@ -4,6 +4,8 @@ Description=inference processor [Service] ExecStart=/home/sdt/miniconda3/bin/python /home/sdt/Workspace/gseps/rabbitmq_test/inference.py Restart=on-failure +Group=sdt +User=sdt [Install] WantedBy=multi-user.target diff --git a/install.sh b/install.sh index 63a780b..f9b10a6 100644 --- a/install.sh +++ b/install.sh @@ -1,6 +1,20 @@ #!/bin/bash -sudo cp ftp_server.service /etc/systemd/system/ -sudo cp inference_processor.service /etc/systemd/system/ + +if [ -f '/etc/systemd/system/ftp_server.service' ]; then + sudo systemctl disable ftp_server.service + sudo systemctl stop ftp_server.service + sudo rm -rf /etc/systemd/system/ftp_server.service +fi + +if [ -f '/etc/systemd/system/inference_processor.service' ]; then + sudo systemctl disable inference_processor.service + sudo systemctl stop inference_processor.service + sudo rm -rf /etc/systemd/system/inference_processor.service +fi + +sudo cp /home/sdt/Workspace/gseps/ftp_server.service /etc/systemd/system/ +sudo cp /home/sdt/Workspace/gseps/inference_processor.service /etc/systemd/system/ + sudo systemctl daemon-reload sudo systemctl enable ftp_server.service sudo systemctl enable inference_processor.service