텐서플로우
-
TF Keras 학습 속도 줄이기 (30%)라이브러리/Tensorflow keras 2022. 4. 26. 10:15
만약 최신 GPU로 TF Keras를 이용하고 있다면, 몇 줄만 추가해서 30% 정도 학습속도를 줄일 수 있다. from tensorflow.keras import mixed_precision policy = mixed_precision.Policy('mixed_float16') mixed_precision.set_global_policy(policy) *주의: TF에선 마지막 레이어가 softmax (sigmoid인 경우도)라면 stability를 위해 float32로 바꾸는 걸 추천함! output_layer = tf.keras.layers.Activation('sigmoid', dtype='float32')(output_layer) 참고 https://www.kaggle.com/competition..
-
tf.data, TFRecord라이브러리/Tensorflow keras 2022. 1. 5. 00:32
tf.data https://www.tensorflow.org/guide/data The tf.data API enables you to build complex input pipelines from simple, reusable pieces. For example, the pipeline for an image model might aggregate data from files in a distributed file system, apply random perturbations to each image, and merge randomly selected images into a batch for training. The pipeline for a text model might involve extrac..
-
Tensorflow_io로 dicom 데이터 다루기컴퓨터 비젼(Computer Vision) 2021. 8. 9. 23:48
pydicom라이브러리를 사용하지 않고 Dicom 파일을 읽고 시각화해보자. 사용 데이터: https://www.kaggle.com/c/rsna-miccai-brain-tumor-radiogenomic-classification/data 1. tensorflow-io 설치 !pip install -q tensorflow-io 2. 필수 라이브러리 호출 import matplotlib.pyplot as plt import numpy as np import tensorflow as tf import tensorflow_io as tfio import glob import os 3. 파일 읽어오기 train_dir = '../input/rsna-miccai-brain-tumor-radiogenomic-clas..