Initial working version of VOC XML import

README.md

@@ -104,7 +104,7 @@ You can find examples of export files along with a description and schema on our
 |:-------------:|:---:|:----:|:-------:|:--------:|:---------:|:----------:|
 | **Point** | ☐ | ✗ | ☐ | ☐ | ☐ | ✗ |
 | **Line** | ☐ | ✗ | ✗ | ✗ | ✗ | ✗ |
-| **Rect** | ☐ | ✓ | ☐ | ☐ | ✓ | ✗ |
+| **Rect** | ☐ | ✓ | ✓ | ☐ | ✓ | ✗ |
 | **Polygon** | ☐ | ✗ | ☐ | ☐ | ✓ | ☐ |
 | **Label** | ☐ | ✗ | ✗ | ✗ | ✗ | ✗ |
 

src/data/ImportFormatData.ts

@@ -13,6 +13,10 @@ export const ImportFormatData: ImportFormatDataMap = {
  {
  type: AnnotationFormatType.YOLO,
  label: 'Multiple files in YOLO format along with labels names definition - labels.txt file.'
+ },
+ {
+ type: AnnotationFormatType.VOC,
+ label: 'Multiple files in VOC XML format.'
  }
  ],
  [LabelType.POINT]: [],

src/data/ImporterSpecData.ts

@@ -2,6 +2,7 @@ import {AnnotationFormatType} from './enums/AnnotationFormatType';
 import {AnnotationImporter} from '../logic/import/AnnotationImporter';
 import {COCOImporter} from '../logic/import/coco/COCOImporter';
 import {YOLOImporter} from '../logic/import/yolo/YOLOImporter';
+import {VOCImporter} from '../logic/import/voc/VOCImporter';
 
 export type ImporterSpecDataMap = Record<AnnotationFormatType, typeof AnnotationImporter>;
 
@@ -11,6 +12,6 @@ export const ImporterSpecData: ImporterSpecDataMap = {
  [AnnotationFormatType.CSV]: undefined,
  [AnnotationFormatType.JSON]: undefined,
  [AnnotationFormatType.VGG]: undefined,
- [AnnotationFormatType.VOC]: undefined,
+ [AnnotationFormatType.VOC]: VOCImporter,
  [AnnotationFormatType.YOLO]: YOLOImporter
 }

src/data/enums/AcceptedFileType.ts

@@ -1,5 +1,6 @@
 export enum AcceptedFileType {
  IMAGE = 'image/jpeg, image/png',
  TEXT = 'text/plain',
- JSON = 'application/json'
+ JSON = 'application/json',
+ XML = 'application/xml',
 }

src/logic/import/voc/VOCImporter.ts

@@ -0,0 +1,105 @@
+import {ImageData, LabelName, LabelRect} from '../../../store/labels/types';
+import {LabelUtil} from "../../../utils/LabelUtil";
+import {AnnotationImporter} from '../AnnotationImporter';
+import {LabelsSelector} from '../../../store/selectors/LabelsSelector';
+
+type FileParseResult = {
+ filename: string,
+ labeledBoxes: LabelRect[]
+};
+
+type VOCImportResult = {
+ labelNames: Record<string, LabelName>,
+ fileParseResults: FileParseResult[],
+};
+
+export class VOCImporter extends AnnotationImporter {
+ public import(
+ filesData: File[],
+ onSuccess: (imagesData: ImageData[], labelNames: LabelName[]) => any,
+ onFailure: (error?:Error) => any
+ ): void {
+ try {
+ const inputImagesData: Record<string, ImageData> = VOCImporter.mapImageData();
+
+ this.loadAndParseFiles(filesData).then(results => {
+ for (const result of results.fileParseResults) {
+ if (inputImagesData[result.filename]) {
+ inputImagesData[result.filename].labelRects = result.labeledBoxes;
+ }
+ }
+
+ onSuccess(
+ Array.from(Object.values(inputImagesData)),
+ Array.from(Object.values(results.labelNames))
+ );
+ }).catch((error: Error) => onFailure(error));
+ } catch (error) {
+ onFailure(error as Error)
+ }
+ }
+
+ private loadAndParseFiles(files: File[]): Promise<VOCImportResult> {
+ const parser = new DOMParser();
+
+ return Promise.all(files.map(file => file.text())).then(textFiles => 
+ textFiles.reduce((current, fileData) => 
+ VOCImporter.parseDocumentIntoImageData(parser.parseFromString(fileData, 'application/xml'), current), 
+ {
+ labelNames: {},
+ fileParseResults: [],
+ } as VOCImportResult)
+ );
+ }
+
+ protected static parseDocumentIntoImageData(document: Document, { fileParseResults, labelNames }: VOCImportResult): VOCImportResult {
+ const root = document.getElementsByTagName('annotation')[0];
+ const filename = root.getElementsByTagName('filename')[0].textContent;
+
+ const [labeledBoxes, newLabelNames] = this.parseAnnotationsFromFileString(document, labelNames);
+
+ return {
+ labelNames: newLabelNames,
+ fileParseResults: fileParseResults.concat({
+ filename,
+ labeledBoxes
+ }),
+ };
+ }
+
+ protected static parseAnnotationsFromFileString(document: Document, labelNames: Record<string, LabelName>): 
+ [LabelRect[], Record<string, LabelName>] {
+ const newLabelNames: Record<string, LabelName> = Object.assign(labelNames);
+ return [Array.from(document.getElementsByTagName('object')).map(d => {
+ const labelName = d.getElementsByTagName('name')[0].textContent;
+ const bbox = d.getElementsByTagName('bndbox')[0];
+ const xmin = parseInt(bbox.getElementsByTagName('xmin')[0].textContent);
+ const xmax = parseInt(bbox.getElementsByTagName('xmax')[0].textContent);
+ const ymin = parseInt(bbox.getElementsByTagName('ymin')[0].textContent);
+ const ymax = parseInt(bbox.getElementsByTagName('ymax')[0].textContent);
+ const rect = {
+ x: xmin,
+ y: ymin,
+ height: ymax - ymin,
+ width: xmax - xmin, 
+ };
+
+ if (!newLabelNames[labelName]) {
+ newLabelNames[labelName] = LabelUtil.createLabelName(labelName);
+ }
+
+ const labelId = newLabelNames[labelName].id;
+
+ return LabelUtil.createLabelRect(labelId, rect);
+ }), newLabelNames];
+ }
+
+ private static mapImageData(): Record<string, ImageData> {
+ return LabelsSelector.getImagesData().reduce(
+ (c: Record<string, ImageData>, i: ImageData) => {
+ c[i.fileData.name] = i;
+ return c;
+ }, {}
+ );
+ }
+}

src/views/PopupView/ImportLabelPopup/ImportLabelPopup.tsx

@@ -64,7 +64,8 @@ const ImportLabelPopup: React.FC<IProps> = (
  const { getRootProps, getInputProps } = useDropzone({
  accept: {
  "application/json": [".json" ],
- "text/plain": [".txt"]
+ "text/plain": [".txt"],
+ "application/xml": [".xml"],
  },
  multiple: true,
  onDrop: (acceptedFiles) => {