2017-11-16 11 views
0

私はTFRコードをロードしていますが、私は6種類のクラスと合計約8000の画像を持っています。ここでtf.train.shuffle_batchランダムな画像を生成しないようです。

images, labels = tf.train.shuffle_batch([image, label], batch_size=10, 
capacity=3000, num_threads=3, min_after_dequeue=2000) 

は私が生産していますイメージされています:あなたがラベルで見ることができるように、私はのみから画像を

10, 299, 299, 3) 
[1 0 1 1 1 1 1 1 0 0] 
(10, 299, 299, 3) 
[2 2 0 2 0 2 1 0 0 2] 
(10, 299, 299, 3) 
[0 1 1 1 2 2 2 2 1 2] 
(10, 299, 299, 3) 
[0 0 2 0 2 2 0 1 1 0] 
(10, 299, 299, 3) 
[0 2 0 0 2 0 0 2 1 2] 

を取得しています。ここ

は私の現在の引数がどのように見えるかです最初の3つは クラスです。

容量とmin_after_dequeueで使用する特定の値または比率がありますか? (私は実際に両方の値を増額しようとしたが、私のPCはかなり遅くなるように見えた、私は再起動し、再試行してくださいする必要があるかもしれません)

ありがとう

答えて

0

TensorFlow API webpage

min_after_dequeueを言う:キュー内の最小数の要素デキューの後、要素のレベルの混合を保証するために使用されます。

capacity:キュー内の最大要素数。

したがって、min_after_dequeue(たとえば8000)を増やして、capacitymin_after_dequeue以上であることを確認してください。

関連する問題