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
name: "PVANET" | |
input: "data" | |
input_shape { dim: 1 dim: 3 dim: 512 dim: 512 } | |
input: "fcn_th" | |
input_shape { dim: 1 dim: 1 } | |
################################################################################ | |
## Convolution |
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
input: "decoder" | |
input_shape { dim: 1 dim: 1 dim: 256 } | |
input: "sample_gt_cont" | |
input_shape { dim: 1 dim: 1 } | |
input: "sample_gt_label_input" | |
input_shape { dim: 1 dim: 1 } | |
layer { |
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
name : "DECODER" | |
input: "sample_gt_label_input" | |
input_shape { dim: 1 dim: 1 } | |
input2: "sample_gt_cont" | |
input_shape { dim: 1 dim: 1 } | |
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
name: "LSTM" | |
input: "data" | |
input_shape { dim: 320 dim: 1 } | |
input: "clip" | |
input_shape { dim: 320 dim: 1 } | |
input: "label" | |
input_shape { dim: 320 dim: 1 } | |
layer { | |
name: "Silence" | |
type: "Silence" |
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
import os | |
import traceback | |
import time | |
import xml.etree.ElementTree as ET | |
import torch | |
from torch.utils.data import Dataset | |
from data.utils import filter_by_shape, extend_bbox, image_channel_mean, bbox_to_quad | |
from data.transforms import TransformCrop, TransformPad, BinarizerWeightGeneration |
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
cat /usr/local/cuda/version.txt | |
nvcc --version | |
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 | |
or | |
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2 |
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
Notes on linking with C/C++ libraries with gcc/g++ when building an | |
existing project. | |
-- Iain Murray, 2015. | |
There are at least three things that can go wrong when trying to link | |
with a library that is in a non-standard location on your machine: | |
1. The compiler can't find the .h header files. | |
2. The linker can't find the library's binary .a or .so files. |
OlderNewer