Skip to content

Commit

Permalink
CTPN verified working on CPU. Update dockerfiles, next GPU and remova…
Browse files Browse the repository at this point in the history
…l of caffe dockerfiles.

Add tutorial using client API. Improve documentation, split cloud into AWS/Heroku and GCP with Kubernetes example.
Next create an intermediate tutorial using COCO dataset. Leverage ability to import region annotations specified in
JSON file with filenames and frame_index used as invariant identifiers.  Streamlining creation of pre-indexed datasets for
LFW, COCO and YFCC subsets. Also add ability to import from a local dir /ingest.

Next:
1. Reduce use of dvaapp in dvaui. Eventually ensuring that everything goes through the REST API.
2. improve paper & documentation
  • Loading branch information
akshay bhat committed Nov 27, 2017
1 parent 596b92b commit fb3db3c
Show file tree
Hide file tree
Showing 6 changed files with 25 additions and 2 deletions.
1 change: 1 addition & 0 deletions .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ install: "pip install -q -r requirements.txt > /dev/null"

before_script:
- cd repos/lopq/python && python setup.py install && cd ../../..
- cd repos/tf_ctpn_cpu/lib/utils && ./make.sh && cd ../../../..
- python manage.py migrate
- python manage.py runserver 8000 &
- cd tests/ci && wget --quiet https://www.dropbox.com/s/t4bgkh2w2ow8o10/WorldIsNotEnough.mp4 && cd ../..
Expand Down
16 changes: 16 additions & 0 deletions configs/models.json
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,22 @@
"algorithm": "CTPN",
"mode": "C"
},
{
"model_type": "detector",
"name": "tf_textbox",
"url": "https://www.dropbox.com/s/ghxowdua65r82d6/checkpoint",
"filename": "checkpoint",
"additional_files":[
{"url": "https://www.dropbox.com/s/v9w4mhcks7a719k/VGGnet_fast_rcnn_iter_50000.ckpt.data-00000-of-00001",
"filename": "VGGnet_fast_rcnn_iter_50000.ckpt.data-00000-of-00001"},
{"url": "https://www.dropbox.com/s/2d0licy8npype6r/VGGnet_fast_rcnn_iter_50000.ckpt.index",
"filename": "VGGnet_fast_rcnn_iter_50000.ckpt.index"},
{"url": "https://www.dropbox.com/s/vwmz2ra9jovlxjd/VGGnet_fast_rcnn_iter_50000.ckpt.meta",
"filename": "VGGnet_fast_rcnn_iter_50000.ckpt.meta"}
],
"algorithm": "TFCTPN",
"mode": "T"
},
{
"model_type": "detector",
"name": "face",
Expand Down
2 changes: 2 additions & 0 deletions deploy/dockerfiles/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,8 @@ RUN pip install torchvision
WORKDIR "/root/DVA"
WORKDIR "/root/DVA/repos/lopq/python"
RUN python setup.py install
WORKDIR "/root/DVA/repos/tf_ctpn_cpu/lib/utils/"
RUN ./make.sh
WORKDIR "/root/DVA"
VOLUME ["/root/DVA/dva/media"]
EXPOSE 80
2 changes: 2 additions & 0 deletions deploy/dockerfiles/Dockerfile.gpu
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,8 @@ RUN pip install --no-deps h5py
WORKDIR "/root/DVA"
WORKDIR "/root/DVA/repos/lopq/python"
RUN python setup.py install
WORKDIR "/root/DVA/repos/tf_ctpn_cpu/lib/utils/"
RUN ./make.sh
WORKDIR "/root/DVA"
VOLUME ["/root/DVA/dva/media"]
EXPOSE 80
3 changes: 2 additions & 1 deletion deploy/dockerfiles/rebuild.sh
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
#!/usr/bin/env bash
set -xe
docker build -t akshayubhat/dva-auto:latest .
docker build -t akshayubhat/dva-auto:latest .
docker build -t akshayubhat/dva-auto:gpu Dockerfile.gpu
3 changes: 2 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -25,4 +25,5 @@ uwsgi
lmdb
gunicorn
django-celery-beat
cython
cython
easydict

0 comments on commit fb3db3c

Please sign in to comment.