コード例 #1
0
  def testSimple(self):
    labels = [9, 3, 0]
    records = [self._record(labels[0], 0, 128, 255),
               self._record(labels[1], 255, 0, 1),
               self._record(labels[2], 254, 255, 0)]
    i = len(records)
    contents = b"".join([record for record, _ in records])
    expected = [expected for _, expected in records]
    filename = os.path.join('/tmp', "cifar_test.bin")
    open(filename, "wb").write(contents)

    with self.test_session() as sess:
      q = tf.FIFOQueue(99, [tf.string], shapes=())
      q.enqueue([filename]).run()
      q.close().run()
      result = cifar10_input.read_cifar10(q)

      for i in range(3):
        key, label, uint8image = sess.run([
            result.key, result.label, result.uint8image])
        print("the value of key"+key)
        # print(label)
        # print(uint8image)
        # print(expected[i])
        self.assertEqual("%s:%d" % (filename, i), compat.as_text(key))
        self.assertEqual(labels[i], label)
        self.assertAllEqual(expected[i], uint8image)

      with self.assertRaises(tf.errors.OutOfRangeError):
        sess.run([result.key, result.uint8image])
コード例 #2
0
    def testSimple(self):
        labels = [9, 3, 0]
        records = [
            self._record(labels[0], 0, 128, 255),
            self._record(labels[1], 255, 0, 1),
            self._record(labels[2], 254, 255, 0)
        ]
        i = len(records)
        contents = b"".join([record for record, _ in records])
        expected = [expected for _, expected in records]
        filename = os.path.join('/tmp', "cifar_test.bin")
        open(filename, "wb").write(contents)

        with self.test_session() as sess:
            q = tf.FIFOQueue(99, [tf.string], shapes=())
            q.enqueue([filename]).run()
            q.close().run()
            result = cifar10_input.read_cifar10(q)

            for i in range(3):
                key, label, uint8image = sess.run(
                    [result.key, result.label, result.uint8image])
                print("the value of key" + key)
                # print(label)
                # print(uint8image)
                # print(expected[i])
                self.assertEqual("%s:%d" % (filename, i), compat.as_text(key))
                self.assertEqual(labels[i], label)
                self.assertAllEqual(expected[i], uint8image)

            with self.assertRaises(tf.errors.OutOfRangeError):
                sess.run([result.key, result.uint8image])
コード例 #3
0
    def testSimple2(self):
        # url = '/home/aurora/hdd/workspace/PycharmProjects/data/aurora2/'
        # labels = [1, 2, 3, 4]
        # images = ['N20031221G035621.bmp', 'N20031221G031111.bmp', 'N20031221G101531.bmp', 'N20031222G041721.bmp']
        #
        # results, expectes = self.img2list(url, labels, images)
        # i = len(results)
        # contents = b"".join(results)
        # expected = [expected for expected in expectes]
        # filename = os.path.join('/tmp', "aurora_test.bin")
        # open(filename, "wb").write(contents)

        labels = 3
        expected, nums = self.gen_bin_files(
            '/home/aurora/hdd/workspace/PycharmProjects/data/aurora2/')
        # filename = '/home/aurora/hdd/workspace/PycharmProjects/data/aurora2/bin_files/type1_data21.bin'
        filename = '/home/aurora/hdd/workspace/PycharmProjects/data/aurora2/bin_files_128/type3_data03.bin'
        with self.test_session() as sess:
            q = tf.FIFOQueue(10000, [tf.string], shapes=())
            q.enqueue([filename]).run()
            q.close().run()
            result = input_data.read_aurora(q)

            for i in range(nums):
                key, label, uint8image = sess.run(
                    [result.key, result.label, result.uint8image])
                # print("the value of key"+key)
                print(label)
                # print(len(uint8image), len(uint8image[0]), len(uint8image[0][1]))
                # print(uint8image)
                # print(expected[i])
                self.assertEqual("%s:%d" % (filename, i), compat.as_text(key))
                self.assertEqual(labels, label)
                self.assertAllEqual(expected[i], uint8image)

            with self.assertRaises(tf.errors.OutOfRangeError):
                sess.run([result.key, result.uint8image])
コード例 #4
0
ファイル: input_data_test.py プロジェクト: xuptlib/annet
  def testSimple2(self):
    # url = '/home/aurora/hdd/workspace/PycharmProjects/data/aurora2/'
    # labels = [1, 2, 3, 4]
    # images = ['N20031221G035621.bmp', 'N20031221G031111.bmp', 'N20031221G101531.bmp', 'N20031222G041721.bmp']
    #
    # results, expectes = self.img2list(url, labels, images)
    # i = len(results)
    # contents = b"".join(results)
    # expected = [expected for expected in expectes]
    # filename = os.path.join('/tmp', "aurora_test.bin")
    # open(filename, "wb").write(contents)

    labels = 3
    expected, nums = self.gen_bin_files('/home/aurora/hdd/workspace/PycharmProjects/data/aurora2/')
    # filename = '/home/aurora/hdd/workspace/PycharmProjects/data/aurora2/bin_files/type1_data21.bin'
    filename = '/home/aurora/hdd/workspace/PycharmProjects/data/aurora2/bin_files_128/type3_data03.bin'
    with self.test_session() as sess:
      q = tf.FIFOQueue(10000, [tf.string], shapes=())
      q.enqueue([filename]).run()
      q.close().run()
      result = input_data.read_aurora(q)

      for i in range(nums):
        key, label, uint8image = sess.run([
            result.key, result.label, result.uint8image])
        # print("the value of key"+key)
        print(label)
        # print(len(uint8image), len(uint8image[0]), len(uint8image[0][1]))
        # print(uint8image)
        # print(expected[i])
        self.assertEqual("%s:%d" % (filename, i), compat.as_text(key))
        self.assertEqual(labels, label)
        self.assertAllEqual(expected[i], uint8image)

      with self.assertRaises(tf.errors.OutOfRangeError):
        sess.run([result.key, result.uint8image])