diff --git a/contrib/HumanSeg/models/humanseg.py b/contrib/HumanSeg/models/humanseg.py index 88bb680d23f695a06fd1c961758bd0a443c5c856..413be8925e62cce21c55fbbf209573e8e0be16f8 100644 --- a/contrib/HumanSeg/models/humanseg.py +++ b/contrib/HumanSeg/models/humanseg.py @@ -25,6 +25,7 @@ import time import tqdm import cv2 import yaml +import shutil import paddleslim as slim import utils @@ -293,7 +294,7 @@ class SegModel(object): save_dir=None, batch_size=1, batch_nums=10, - cache_dir="./.temp", + cache_dir=".temp", quant_type="offline"): if quant_type == "offline": self.arrange_transform(transforms=dataset.transforms, mode='quant') @@ -324,7 +325,7 @@ class SegModel(object): post_training_quantization.quantize() post_training_quantization.save_quantized_model(save_dir) if cache_dir is not None: - os.system('rm -r ' + cache_dir) + shutil.rmtree(cache_dir) else: float_prog, _ = slim.quant.convert( self.test_prog, self.exe.place, save_int8=True)