This file contains hidden or 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. |
This file contains hidden or 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 hidden or 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 hidden or 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 hidden or 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 hidden or 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 hidden or 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 hidden or 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" | |
layer { | |
type: "YOU DATA LAYER" | |
top: "mask_gt" ### 2 * 640 * 640 * 1 ### mask for text/non-text segmentation | |
top: "mask_iou_angle" ### 1 * 640 * 640 * 5 ### iou with 4 channels and angle with one channel | |
top: "gt_bbox" ### N * 8 ### grounding truth boxes for text (for computing loss) | |
top: "ignore_bbox" ### N * 8 ### grounding truth boxes for text (for ignoring) | |
} |
This file contains hidden or 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
void CropAndResizePerBox( | |
const float * image_data, | |
const int batch_size, | |
const int depth, | |
const int image_height, | |
const int image_width, | |
const float * boxes_data, | |
const int * box_index_data, | |
const int start_box, |
This file contains hidden or 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
/* ex: set ts=4: */ | |
/* | |
based on code from http://www.progressive-coding.com/tutorial.php?id=3#end | |
--------------------------------------------------------------------------- | |
Copyright (c) 2007-2010, Laurent Haan, Luxembourg, LU. All rights reserved. | |
LICENSE TERMS | |
The redistribution and use of this software (with or without changes) | |
is allowed without the payment of fees or royalties provided that: |
NewerOlder