forked from prehistoric-systems/comixify
b5dd5cffc0
* Add keyframe model * Add segmentation utils * Add keyframes extraction pipeline * Add keyframe tests * Update dockerfile to include caffe * Add summe pretrained model * Add video for testing * Update keyframe pipeline, tests * Update settings to use in memory db for tests * Set keyframe number to 10, fix bugs * Fix keyframe order * Make requested changes * Fix Dockerfile * Make requested changes * Make requested changes * Add blank lines * Change dockerfile base cuda image to devel version * Add modified Cuda.cmake for Dockerfile * Add pyyaml dependecy to dockerfile * Update dockerfile * Update dockerfile * Fix markdown version error * Fix markdown version error * Change caffe installation to make * Update dockerfile * Update dockerfile * Fix boost imoprt * Fix boost not found bug * Add feature normalisation * Fix dateutil, fix caffe root slash * Fix slash bug * Add batching to feature extraction * Add model caching to keyframes extraction * Fix output images to be in proper range * Add time logging * Change feature batch to 128 * Change dockerfile * Fix dockerfile * Change feature batch to 10 * Add set mode gpu * Change feature batch to 64 * Change feature batch to 32 * Add I-frame frame sampling * Cleanup * Delete Cuda.cmake * Remove comments from Makefile.config * Cleanup * Fix color scheme switching * Remove cudnn.hpp, change caffe to 1.0 * Remove cudnn.hpp copy in dockerfile * Remove redundant run's in dockerfile * Change pretrained model
21 lines
No EOL
824 B
Text
21 lines
No EOL
824 B
Text
USE_CUDNN := 1
|
|
CUDA_DIR := /usr/local/cuda
|
|
|
|
CUDA_ARCH := -gencode arch=compute_30,code=sm_30 \
|
|
-gencode arch=compute_35,code=sm_35 \
|
|
-gencode arch=compute_50,code=sm_50 \
|
|
-gencode arch=compute_52,code=sm_52 \
|
|
-gencode arch=compute_60,code=sm_60 \
|
|
-gencode arch=compute_61,code=sm_61 \
|
|
-gencode arch=compute_61,code=compute_61
|
|
BLAS := atlas
|
|
PYTHON_LIBRARIES := boost_python3 python3.6m
|
|
PYTHON_INCLUDE := /usr/include/python3.6 \
|
|
/usr/local/lib/python3.6/dist-packages/numpy/core/include
|
|
PYTHON_LIB := /usr/lib
|
|
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
|
|
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
|
|
BUILD_DIR := build
|
|
DISTRIBUTE_DIR := distribute
|
|
TEST_GPUID := 0
|
|
Q ?= @ |