Differences
This shows you the differences between two versions of the page.
Both sides previous revisionPrevious revisionNext revision | Previous revisionNext revisionBoth sides next revision |
tensorflow:fast-style-transfer [2018/04/22] – gomida | tensorflow:fast-style-transfer [2018/04/22] – gomida |
---|
<sq tensorflow/fast-style-transfer/r.jpg>기본 옵션으로 학습한 모델</sq> | <sq tensorflow/fast-style-transfer/r.jpg>기본 옵션으로 학습한 모델</sq> |
<sq tensorflow/fast-style-transfer/r_original.jpg>제공되는 wave.ckpt 모델</sq> | <sq tensorflow/fast-style-transfer/r_original.jpg>제공되는 wave.ckpt 모델</sq> |
| \\ |
| ---- |
=== TF Lite === | === TF Lite === |
스트라이드를 줄여 디컨볼루션을 제거하고 러닝하면 TransposeConv는 제거할 수 있지만 모델 사이즈 제한에 걸린다. | 스트라이드를 줄여 디컨볼루션을 제거하고 학습하면 TransposeConv는 제거할 수 있지만 모델 사이즈 제한에 걸린다. 버전 1.7.0 toco에서는 에러 없이 변환되지만 런타임 interpreter->AllocateTensors() 호출 시 에러가 발생한다. |
| <code> |
| Failed to allocate tensors! |
| tensorflow/contrib/lite/kernels/div.cc:48 NumDimensions(input1) != NumDimensions(input2) (4 != 0) |
| </code> |
<code> | <code> |
gomida@TITAN:~/github/tensorflow$ ./bazel-bin/tensorflow/contrib/lite/toco/toco --input_file=/home/gomida/github/tensorflow-fast-style-transfer/frozen_out3.pb --output_file=/home/gomida/github/tensorflow-fast-style-transfer/frozen_out3_r16.tflite --input_array=input --output_array=Tanh --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE | gomida@TITAN:~/github/tensorflow$ ./bazel-bin/tensorflow/contrib/lite/toco/toco --input_file=/home/gomida/github/tensorflow-fast-style-transfer/frozen_out3.pb --output_file=/home/gomida/github/tensorflow-fast-style-transfer/frozen_out3_r16.tflite --input_array=input --output_array=Tanh --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE |
Aborted (core dumped) | Aborted (core dumped) |
</code> | </code> |
| <sq tensorflow/fast-style-transfer/r_out3.jpg>스트라이드를 줄이고 후반 디컨볼루션을 제거한 모델</sq> |