-
Notifications
You must be signed in to change notification settings - Fork 3.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Increase request body size for TransT tracker (#6073)
- Loading branch information
Showing
4 changed files
with
92 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |