Skip to content

Commit

Permalink
Increase request body size for TransT tracker (#6073)
Browse files Browse the repository at this point in the history
  • Loading branch information
azhavoro authored Apr 26, 2023
1 parent 269545b commit 091a035
Show file tree
Hide file tree
Showing 4 changed files with 92 additions and 5 deletions.
3 changes: 2 additions & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
- TDB

### Fixed
- TDB
- Tracking of multiple objects (30 and more) with TransT tracker
(<https://github.com/opencv/cvat/pull/6073>)

### Security
- TDB
Expand Down
5 changes: 3 additions & 2 deletions serverless/pytorch/dschoerk/transt/nuclio/function-gpu.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,8 @@ spec:
kind: 'http'
workerAvailabilityTimeoutMilliseconds: 10000
attributes:
maxRequestBodySize: 33554432 # 32MB
# Set value from the calculation of tracking of 100 objects at the same time on a 4k image
maxRequestBodySize: 268435456 # 256MB

resources:
limits:
Expand All @@ -76,4 +77,4 @@ spec:
restartPolicy:
name: always
maximumRetryCount: 3
mountMode: volume
mountMode: volume
5 changes: 3 additions & 2 deletions serverless/pytorch/dschoerk/transt/nuclio/function.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -65,11 +65,12 @@ spec:
kind: 'http'
workerAvailabilityTimeoutMilliseconds: 10000
attributes:
maxRequestBodySize: 33554432 # 32MB
# Set value from the calculation of tracking of 100 objects at the same time on a 4k image
maxRequestBodySize: 268435456 # 256MB

platform:
attributes:
restartPolicy:
name: always
maximumRetryCount: 3
mountMode: volume
mountMode: volume
84 changes: 84 additions & 0 deletions serverless/pytorch/saic-vul/hrnet/nuclio/function.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,84 @@
metadata:
name: pth-saic-vul-hrnet
namespace: cvat
annotations:
name: HRNET
version: 2
type: interactor
spec:
framework: pytorch
min_pos_points: 1
min_neg_points: 0
animated_gif: https://raw.githubusercontent.com/opencv/cvat/develop/site/content/en/images/hrnet_example.gif
help_message: The interactor allows to get a mask for an object using positive points, and negative points

spec:
description: HRNet18 for click based interactive segmentation
runtime: 'python:3.8'
handler: main:handler
eventTimeout: 30s
env:
- name: PYTHONPATH
value: /opt/nuclio/hrnet

build:
image: cvat-pth.saic-vul.hrnet
baseImage: ubuntu:20.04

directives:
preCopy:
- kind: ENV
value: DEBIAN_FRONTEND=noninteractive
- kind: RUN
value: apt-get update && apt-get install software-properties-common -y
- kind: RUN
value: add-apt-repository ppa:deadsnakes/ppa
- kind: RUN
value: apt remove python* -y
- kind: RUN
value: apt-get update && apt-get install -y --no-install-recommends build-essential git curl libglib2.0-0 software-properties-common python3 python3.6-dev python3-pip python3-tk
- kind: RUN
value: ln -s /usr/bin/pip3 /usr/local/bin/pip && ln -s /usr/bin/python3 /usr/bin/python
- kind: RUN
value: pip3 install --upgrade pip
- kind: WORKDIR
value: /opt/nuclio
- kind: RUN
value: git clone https://github.com/saic-vul/ritm_interactive_segmentation.git hrnet
- kind: WORKDIR
value: /opt/nuclio/hrnet
- kind: RUN
value: apt-get install -y --no-install-recommends wget
- kind: RUN
value: wget https://github.com/saic-vul/ritm_interactive_segmentation/releases/download/v1.0/coco_lvis_h18_itermask.pth
- kind: RUN
value: pip3 install setuptools
- kind: RUN
value: pip3 install -r requirements.txt
- kind: RUN
value: apt update && apt install -y libgl1-mesa-glx
- kind: RUN
value: pip3 uninstall torch torch vision -y
- kind: RUN
value: pip install torch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
- kind: WORKDIR
value: /opt/nuclio

triggers:
myHttpTrigger:
maxWorkers: 1
kind: 'http'
workerAvailabilityTimeoutMilliseconds: 10000
attributes:
maxRequestBodySize: 33554432 # 32MB

platform:
attributes:
restartPolicy:
name: always
maximumRetryCount: 3
mountMode: volume

resources:
limits:
cpu: 2

0 comments on commit 091a035

Please sign in to comment.