gameui/ai
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

下載圖像

Browse Source
This commit is contained in:
桜華
2023-05-27 16:40:04 +08:00
parent 788f166909
commit 0966a3c83e
7 changed files with 203 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
test.sh
View File

@@ -2,6 +2,7 @@
# 記錄開始時間戳
start_time=$(date +%s)
rm -f data/sqlite3.db
# 流程測試, 啓動服務, 設定進程名 go_test
go run main.go -procname go_test &
@@ -48,7 +49,7 @@ echo "dataset_id: $dataset_id"
# 修改數據集, images 中增加 url (PATCH /api/datasets/:id)
response=$(curl -X PATCH -H "Content-Type: application/json" -d '{"images":["https://www.google.com/images/branding/googlelogo/2x/googlelogo_color_272x92dp.png"]}' -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/datasets/$dataset_id)
response=$(curl -X PATCH -H "Content-Type: application/json" -d '{"images":["https://img.gameui.net/article-7258-1677745322000@1x456.webp","https://img.gameui.net/article-6477-1682109454000@1x456.webp"]}' -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/datasets/$dataset_id)
[[ ${response: -3} -eq 200 ]] && { echo "修改數據集成功: ${response%???}"; } || exit_service "修改數據集失敗: ${response%???}"
@@ -72,22 +73,27 @@ echo "model_id: $model_id"
#[[ ${response: -3} -eq 200 ]] && { echo "獲取模型訓練進度成功: ${response%???}"; } || exit_service "獲取模型訓練進度失敗: ${response%???}"
# 循環獲取模型訓練進度, 直到訓練完成
while true; do
# 獲取模型訓練進度 (GET /api/models/:id)
response=$(curl -X GET -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/models/$model_id)
[[ ${response: -3} -eq 200 ]] && { echo "獲取模型訓練進度成功: ${response%???}"; } || exit_service "獲取模型訓練進度失敗: ${response%???}"
## 循環獲取模型訓練進度, 直到訓練完成
#while true; do
# # 獲取模型訓練進度 (GET /api/models/:id)
# response=$(curl -X GET -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/models/$model_id)
# [[ ${response: -3} -eq 200 ]] && { echo "獲取模型訓練進度成功: ${response%???}"; } || exit_service "獲取模型訓練進度失敗: ${response%???}"
#
# # 取出進度字段的值, 值爲 int
# progress=$(echo "${response%???}" | grep -o '"progress": [0-9]*' | awk '{print $2}')
# echo "progress: $progress"
#
# # 如果進度爲 100, 訓練完成, 跳出循環
# [[ $progress -eq 100 ]] && { echo "訓練完成"; break; }
#
# # 休眠 5 秒
# sleep 5
#done
# 取出進度字段的值, 值爲 int
progress=$(echo "${response%???}" | grep -o '"progress": [0-9]*' | awk '{print $2}')
echo "progress: $progress"
# 如果進度爲 100, 訓練完成, 跳出循環
[[ $progress -eq 100 ]] && { echo "訓練完成"; break; }
# 休眠 5 秒
sleep 5
done
# 服務器列表
response=$(curl -X GET -b "session_id=$session_id" -s -w "%{http_code}" http://localhost:8080/api/servers)
[[ ${response: -3} -eq 200 ]] && { echo "獲取服務器列表成功: ${response%???}"; } || exit_service "獲取服務器列表失敗: ${response%???}"
sleep 10
exit_service "測試結束, 全部通過, 用費時間: $(($(date +%s) - $start_time)) 秒"