Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/keras/src/datasets/cifar.py: 23%

1# Copyright 2015 The TensorFlow Authors. All Rights Reserved. 

2# 

3# Licensed under the Apache License, Version 2.0 (the "License"); 

4# you may not use this file except in compliance with the License. 

5# You may obtain a copy of the License at 

6# 

7# http://www.apache.org/licenses/LICENSE-2.0 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, 

11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

12# See the License for the specific language governing permissions and 

13# limitations under the License. 

14# ============================================================================== 

15"""Utilities common to CIFAR10 and CIFAR100 datasets.""" 

16 

17import _pickle as cPickle 

18 

19 

20def load_batch(fpath, label_key="labels"): 

21 """Internal utility for parsing CIFAR data. 

22 

23 Args: 

24 fpath: path the file to parse. 

25 label_key: key for label data in the retrieve 

26 dictionary. 

27 

28 Returns: 

29 A tuple `(data, labels)`. 

30 """ 

31 with open(fpath, "rb") as f: 

32 d = cPickle.load(f, encoding="bytes") 

33 # decode utf8 

34 d_decoded = {} 

35 for k, v in d.items(): 

36 d_decoded[k.decode("utf8")] = v 

37 d = d_decoded 

38 data = d["data"] 

39 labels = d[label_key] 

40 

41 data = data.reshape(data.shape[0], 3, 32, 32) 

42 return data, labels 

43