zer0-1s / AutoUrl

AutoUrl

torchserve与url检测联动

steps

python url.py # generate url.pt
mkdir model_store

torch-model-archiver --model-name url \--version 1.0 \--serialized-file url.pt \--extra-files ./index_to_name.json,./MyHandler.py \--handler my_handler.py  \--export-path model_store -f

torchserve --start --model-store model_store --models url=url.mar

dokcer

docker pull pytorch/torchserve

docker run --rm --shm-size=1g \
        --ulimit memlock=-1 \
        --ulimit stack=67108864 \
        -p8080:8080 \
        -p8081:8081 \
        -p8082:8082 \
        -p7070:7070 \
        -p7071:7071 \
        --mount type=bind,source=/home/test/model_store,target=/tmp/models pytorch/torchserve:latest  torchserve --model-store=/tmp/models -

test

curl -X POST "http://localhost:8081/models?url=url.mar" # upload models

curl "http://localhost:8081/models" # check models

curl -v -X PUT "http://localhost:8081/models/url?min_worker=2

curl http://127.0.0.1:8080/predictions/url -T test.txt # no  response

reference:

https://aijishu.com/a/1060000000136517#item-3

https://github.com/FrancescoSaverioZuppichini/torchserve-tryout

https://github.com/zer0-1s/MaliciousURLs/blob/master/train_pickle.py

注：MyHandler.py参考第1,2,3个链接