Skip to content
Snippets Groups Projects
demo.ipynb 170 KiB
Newer Older
    "img.show()\n",
    "\n",
    "\n",
    "class GifReporter():\n",
    "    def __init__(self, dest: str, frames: List = None) -> None:\n",
    "        self.gif = None\n",
    "        self.dest = dest\n",
    "        self.frames = frames\n",
    "\n",
    "    def report(self, res: UQEvalResults, filter: dict,\n",
    "               save = False, split_num: int = 0, \n",
    "               method1: str = \"Empirical DE-CCP\",\n",
    "               method2: str = \"Max additive\") -> Any:  # Gif\n",
    "        filter.update({\"split_num\": split_num})\n",
    "        self.current_results = res\n",
    "        if self.frames is None:\n",
    "            self.random_frames(filter, [method1, method2])\n",
    "        else:\n",
    "            self.find_frames(filter, [method1, method2])\n",
    "        self.draw_bounding_boxes()\n",
    "        self.save_gif() if save else None\n",
    "        return self.gif\n",
    "    \n",
    "    def random_frames(self, filter: dict, methods: List, \n",
    "                      num_frames: int = 10) -> None:\n",
    "        i, self.frames, assigned = 0, [], [] \n",
    "        while len(self.frames) < 10:\n",
    "            fname = None\n",
    "            filter.items() + \n",
    "            if (all(self.current_results[k] == v for k, v in kvs)\n",
    "                and fname not in assigned):\n",
    "                gt_bboxes, conf_intevals1, conf_intevals2 = [None] * 3\n",
    "                self.frames.append({\n",
    "                    \"name\": fname, \"Ground truths\": gt_bboxes,\n",
    "                    methods[0]: conf_intevals1,methods[1]: conf_intevals2})\n",
    "            i += 1 \n",
    "\n",
    "    \n",
    "    def find_frames(self, filter: dict, methods: List) -> None:\n",
    "        pass\n",
    "\n",
    "\n",
    "giffer = GifReporter(dest=\"KITTI_results.gif\")\n",
    "giffer.report(res)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "TBD",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.18"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}