sgkim
/
gseps_inference
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gseps inference server code

This commit is contained in:
ssung 2023-09-01 14:54:57 +09:00
parent f670f846fd
commit 5536ec473d
7 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/ftp_config.json
View File

@ -3,5 +3,5 @@
"ftp_port": 21,
"ftp_id": "sdt",
"ftp_pw": "251327",
"ftp_root_dir": "/home/sdt/Workspace/gseps/rabbitmq_test/inference_result/"
"ftp_root_dir": "/home/sdt/Workspace/gseps_inference/inference_result/"
}

6
config/inference_config.json
View File

@ -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/"
"copied_image_path_from_remote_server": "/home/sdt/Workspace/gseps_inference/image_bucket",
"inference_result_path": "/home/sdt/Workspace/gseps_inference/inference_result/"
}

2
ftp_server.py
View File

@ -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)
# 서버 설정

2
ftp_server.service
View File

@ -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

2
inference.py
View File

@ -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)

2
inference_processor.service
View File

@ -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

18
install.sh
View File

@ -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