Tensorflow custom dataset 만들기

주의할 점

generator must be python callable 에러에 주의할 것

왠만하면 야매로 generator 만들지 말고 class로 만들어서 쓰도록 하자

대용량 훈련 데이터 처리 - Generator로 TF Dataset 만들기

1. 너무 큰 데이터 tf.data.Dataset는 Tensorflow의 훈련 데이터를 다룰 때 참 편리합니다. Padding, Batch, Shuffle, Map 등을 다 지원해주니까요. 일단 데이터를 Dataset으로 변환만 시키면 그다음부터는 아주 편

jins-sw.tistory.com

Creating a TF Dataset using a Data Generator

tf.data.Dataset

medium.com

728x90

FGSM - adversarial training 구현해보기 (0)	2023.06.06
[논문읽기] A combined CNN-LSTM and LSTM-QRNN model for prediction of Idiopathic Pulmonary Fibrosis Progression using CT Scans and Clinical Data (0)	2023.06.01
[논문읽기] Action Recognition in Video Sequences using Deep Bi-Directional LSTM With CNN Features (0)	2023.05.30
[논문읽기] BEIT: BERT Pre-Training of Image Transformers (0)	2023.05.27
[논문읽기] CBAM: Convolutional Block Attention Module (0)	2023.05.27