fastapi-yolo/test_requests_msv3.sh at main · LashaO/fastapi-yolo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/bin/bash

# Test MSV3 model with image URL
echo "Testing with image URL..."
curl -X POST "http://localhost:8000/predict" \
  -H "accept: application/json" \
  -H "Content-Type: application/json" \
  -d '{
    "model_id": "msv3",
    "image_url": "https://upload.wikimedia.org/wikipedia/commons/thumb/a/a3/Green_sea_turtle_%28Chelonia_mydas%29_Moorea.jpg/500px-Green_sea_turtle_%28Chelonia_mydas%29_Moorea.jpg"
  }'

echo "\nTesting with local image path..."
# Test with local image path
curl -X POST "http://localhost:8000/predict" \
  -H "accept: application/json" \
  -H "Content-Type: application/json" \
  -d '{
    "model_id": "msv3",
    "image_path": "/path/to/your/local/image.jpg"
  }'

echo "\nTesting with uploaded file..."
# Test with file upload
curl -X POST "http://localhost:8000/predict" \
  -H "accept: application/json" \
  -F "model_id=msv3" \
  -F "file=@/path/to/your/local/image.jpg"

echo "\nRunning performance test with $N requests..."
URL="http://localhost:8000/predict"
N=${1:-5}  # Default to 5 if no argument provided

time (
  for ((i = 1; i <= N; i++)); do
    echo "Sending request $i..."
    curl -X POST "$URL" \
      -H "accept: application/json" \
      -H "Content-Type: application/json" \
      -d '{"model_id": "msv3", "image_url": "https://upload.wikimedia.org/wikipedia/commons/thumb/a/a3/Green_sea_turtle_%28Chelonia_mydas%29_Moorea.jpg/500px-Green_sea_turtle_%28Chelonia_mydas%29_Moorea.jpg"}' &
  done
  wait
)