4088643824

Build Type

github

Committed by GitHub

Commit Message

Merge 3588a38ba into c1acf7dc8

Pull Request Pull Request #951: build(deps): bump http-cache-semantics from 4.1.0 to 4.1.1 in /packages/cli

Run Details

436 of 530 branches covered (82.26%)

Branch coverage included in aggregate %.

692 of 736 relevant lines covered (94.02%)

4.81 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

80.0

/packages/core/src/dataset-pool/format/coco.ts

import * as DataCook from '@pipcook/datacook';
import { Types as DatasetTypes, ArrayDatasetPoolImpl } from '..';

export type Options = {
  trainAnnotationObj?: DataCook.Dataset.Types.Coco.Meta;
  testAnnotationObj?: DataCook.Dataset.Types.Coco.Meta;
  validAnnotationObj?: DataCook.Dataset.Types.Coco.Meta;
  predictedAnnotationObj?: DataCook.Dataset.Types.Coco.Meta;
};

export const makeDatasetPoolFromCocoFormat = async (
  options: Options
): Promise<
  DatasetTypes.DatasetPool<
    DataCook.Dataset.Types.Sample<DataCook.Dataset.Types.Coco.Image, DataCook.Dataset.Types.Coco.Label>,
    DatasetTypes.Coco.DatasetMeta
  >
> => {
  const train = options.trainAnnotationObj ? DataCook.Dataset.makeDatasetFromCoco(options.trainAnnotationObj) : undefined;
  const test = options.testAnnotationObj ? DataCook.Dataset.makeDatasetFromCoco(options.testAnnotationObj) : undefined;
  const valid = options.validAnnotationObj ? DataCook.Dataset.makeDatasetFromCoco(options.validAnnotationObj) : undefined;
  const predicted = options.predictedAnnotationObj ? DataCook.Dataset.makeDatasetFromCoco(options.predictedAnnotationObj) : undefined;

  const categories = options.trainAnnotationObj ? DataCook.Dataset.extractCategoriesFromCoco(options.trainAnnotationObj) : undefined;

  const datasetMeta: DatasetTypes.Coco.DatasetMeta = {
    type: DataCook.Dataset.Types.DatasetType.Image,
    size: {
      train: (await train?.nextBatch(-1))?.length || 0,
      test: (await test?.nextBatch(-1))?.length || 0,
      valid: (await valid?.nextBatch(-1))?.length || 0,
      predicted: (await predicted?.nextBatch(-1))?.length || 0
    },
    categories,
    info: options.trainAnnotationObj?.info,
    licenses: options.trainAnnotationObj?.licenses
  };
  await Promise.all([
    train?.seek(0),
    test?.seek(0),
    valid?.seek(0),
    predicted?.seek(0)
  ]);
  return ArrayDatasetPoolImpl.from({
    train,
    test,
    valid,
    predicted
  }, datasetMeta);
};

1	import * as DataCook from '@pipcook/datacook';	6✔
2	import { Types as DatasetTypes, ArrayDatasetPoolImpl } from '..';	6✔
3
4	export type Options = {
5	trainAnnotationObj?: DataCook.Dataset.Types.Coco.Meta;
6	testAnnotationObj?: DataCook.Dataset.Types.Coco.Meta;
7	validAnnotationObj?: DataCook.Dataset.Types.Coco.Meta;
8	predictedAnnotationObj?: DataCook.Dataset.Types.Coco.Meta;
9	};
10
11	export const makeDatasetPoolFromCocoFormat = async (	6✔
12	options: Options
13	): Promise<
14	DatasetTypes.DatasetPool<
15	DataCook.Dataset.Types.Sample<DataCook.Dataset.Types.Coco.Image, DataCook.Dataset.Types.Coco.Label>,
16	DatasetTypes.Coco.DatasetMeta
17	>
18	> => {
19	const train = options.trainAnnotationObj ? DataCook.Dataset.makeDatasetFromCoco(options.trainAnnotationObj) : undefined;	3!
20	const test = options.testAnnotationObj ? DataCook.Dataset.makeDatasetFromCoco(options.testAnnotationObj) : undefined;	3!
21	const valid = options.validAnnotationObj ? DataCook.Dataset.makeDatasetFromCoco(options.validAnnotationObj) : undefined;	3✔
22	const predicted = options.predictedAnnotationObj ? DataCook.Dataset.makeDatasetFromCoco(options.predictedAnnotationObj) : undefined;	3!
23
24	const categories = options.trainAnnotationObj ? DataCook.Dataset.extractCategoriesFromCoco(options.trainAnnotationObj) : undefined;	3!
25
26	const datasetMeta: DatasetTypes.Coco.DatasetMeta = {	3✔
27	type: DataCook.Dataset.Types.DatasetType.Image,
28	size: {
29	train: (await train?.nextBatch(-1))?.length \|\| 0,	21!
30	test: (await test?.nextBatch(-1))?.length \|\| 0,	21!
31	valid: (await valid?.nextBatch(-1))?.length \|\| 0,	23✔
32	predicted: (await predicted?.nextBatch(-1))?.length \|\| 0	24!
33	},
34	categories,
35	info: options.trainAnnotationObj?.info,	9!
36	licenses: options.trainAnnotationObj?.licenses	9!
37	};
38	await Promise.all([	3✔
39	train?.seek(0),	9!
40	test?.seek(0),	9!
41	valid?.seek(0),	9✔
42	predicted?.seek(0)	9!
43	]);
44	return ArrayDatasetPoolImpl.from({	3✔
45	train,
46	test,
47	valid,
48	predicted
49	}, datasetMeta);
50	};

alibaba / pipcook / 4088643824

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous