Intelligent Robot

tensorflow2.4.1-gpu 도커 버전에서 모델의 FPS를 체크할 때 발견한 이상현상 (2.7.0-gpu에서도 확인함) data.py 파일 안에 DataGenerator를 상속하여 만든 커스텀 클래스는 아래와 같다 import os import numpy as np from PIL import Image import tensorflow as tf class DataGenerator(tf.keras.utils.Sequence): 'Generates data for Keras' def __init__(self, img_dir, lbl_dir, batch_size=32, shuffle=False): 'Initialization' self.IDs = os.listdir(img_dir) self.IDs.s..

ref. https://www.tensorflow.org/install/source#tested_build_configurations 소스에서 빌드 | TensorFlow Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge 소스에서 빌드 소스에서 TensorFlow pip 패키지를 빌드하고 Ubuntu Linux 및 macOS에 설치합니다. 명령어는 다른 시스템에도 적용될 수 있지만, U www.tensorflow.org

Tensorflow/ Keras Model Profiler Gives you some basic but important information about your tf or keras model like, Model Parameters Model memory requirement on GPU Memory required to store parameters model weights. GPU availability and GPU IDs if available Sample output | Model Profile | Value | Unit | |----------------------------------|---------------|---------| | Selected GPUs | ['0', '1'] | ..