#!/usr/bin/env python
# coding: utf-8

# In[1]:


import numpy as np
import logging
import faster_coco_eval
from faster_coco_eval import COCO, COCOeval_faster

print(f"{faster_coco_eval.__version__=}")

logging.root.setLevel("INFO")
logging.debug("Запись.")


# In[2]:


prepared_coco_in_dict = COCO.load_json("../tests/lvis_dataset/lvis_val_100.json")
prepared_anns = COCO.load_json("../tests/lvis_dataset/lvis_results_100.json")


# In[3]:


iouType = "bbox"


# In[4]:


from lvis import LVISEval

lvis_eval = LVISEval("../tests/lvis_dataset/lvis_val_100.json", "../tests/lvis_dataset/lvis_results_100.json", iouType)


# In[5]:


get_ipython().run_cell_magic('time', '', '\nlvis_eval.evaluate()\n')


# In[6]:


get_ipython().run_cell_magic('time', '', '\nlvis_eval.accumulate()\n')


# In[7]:


get_ipython().run_cell_magic('time', '', '\nlvis_eval.summarize()\n')


# In[8]:


lvis_eval.print_results()


# In[9]:


cocoGt = COCO(prepared_coco_in_dict)
cocoDt = cocoGt.loadRes(prepared_anns)

cocoEval = COCOeval_faster(cocoGt, cocoDt, iouType, lvis_style=True, print_function=print)
cocoEval.params.maxDets = [300]


# In[10]:


get_ipython().run_cell_magic('time', '', '\ncocoEval.evaluate()\n')


# In[11]:


get_ipython().run_cell_magic('time', '', '\ncocoEval.accumulate()\n')


# In[12]:


get_ipython().run_cell_magic('time', '', '\ncocoEval.summarize()\n')