[fix] fix blendcnn and coco import error

Author: ClarkChin08

examples/pytorch/blendcnn/classify.py

@@ -196,13 +196,16 @@ def eval_func(model):
         # print(f"Accuracy: {total_accuracy}")
 
         if args.tune:
-            import lpot
+            from lpot.experimental import Quantization
             # lpot tune
             model.load_state_dict(torch.load(args.input_model))
             dataloader = Bert_DataLoader(loader=data_iter, batch_size=args.batch_size)
 
-            quantizer = lpot.Quantization(args.tuned_yaml)
-            q_model = quantizer(model, q_dataloader=dataloader, eval_func=eval_func)
+            quantizer = Quantization(args.tuned_yaml)
+            quantizer.calib_dataloader = dataloader
+            quantizer.model = model
+            quantizer.eval_func = eval_func
+            q_model = quantizer()
             q_model.save(args.tuned_checkpoint)
 
         elif args.int8:

lpot/experimental/metric/metric.py

@@ -608,7 +608,7 @@ class TensorflowCOCOMAP(BaseMetric):
 
     """
     def __init__(self, anno_path=None):
-        from lpot.metric.coco_label_map import category_map
+        from .coco_label_map import category_map
         if anno_path:
             import os
             import json
@@ -631,7 +631,7 @@ def __init__(self, anno_path=None):
             [cat for cat in self.category_map]) #index
 
     def update(self, predicts, labels, sample_weight=None):
-        from lpot.metric.coco_tools import ExportSingleImageGroundtruthToCoco,\
+        from .coco_tools import ExportSingleImageGroundtruthToCoco,\
             ExportSingleImageDetectionBoxesToCoco
         bbox, str_label,int_label, image_id = labels
         detection = {}
@@ -690,7 +690,7 @@ def reset(self):
         self.annotation_id = 1
 
     def result(self):
-        from lpot.metric.coco_tools import COCOWrapper, COCOEvalWrapper
+        from .coco_tools import COCOWrapper, COCOEvalWrapper
         if len(self.ground_truth_list) == 0:
             logger.warning("sample num is 0 can't calculate mAP")
             return 0