crazysal

name: "PVANET"

input: "data"
input_shape { dim: 1 dim: 3 dim: 512 dim: 512 }
input: "fcn_th"
input_shape { dim: 1 dim: 1 }


################################################################################
## Convolution

crazysal

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 {

crazysal

name : "DECODER"

input: "sample_gt_label_input"
input_shape { dim: 1 dim: 1 }
 
input2: "sample_gt_cont"
input_shape { dim: 1 dim: 1 }
 

crazysal

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"

crazysal

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

crazysal

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

crazysal

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.