blob: 545b8a02bc356f132cb3a41f1a6ed3a1f9c20870 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
#ifndef FLOWNET_DATA_AUGMENTATION_H_
#define FLOWNET_DATA_AUGMENTATION_H_
#include "tensorflow/core/framework/op_kernel.h"
namespace tensorflow {
template<class Device>
void Augment(OpKernelContext *context,
const Device & d,
const int batch_size,
const int channels,
const int src_width,
const int src_height,
const int src_count,
const int out_width,
const int out_height,
const float *src_data,
float *out_data,
const float *transMats,
float *chromatic_coeffs);
} // namespace tensorflow
#endif // FLOWNET_DATA_AUGMENTATION_H_
|
