mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 09:07:10 +08:00
Fix requirements (#59)
* Fix bug in ppyoloe * fix ppyoloe output format * remove some requirements * fix conflicts
This commit is contained in:
@@ -12,11 +12,7 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from tqdm import trange
|
||||
import cv2
|
||||
import numpy as np
|
||||
from .utils import CocoDetection
|
||||
from .utils import COCOMetric
|
||||
import copy
|
||||
import collections
|
||||
|
||||
@@ -27,6 +23,11 @@ def eval_detection(model,
|
||||
conf_threshold=None,
|
||||
nms_iou_threshold=None,
|
||||
plot=False):
|
||||
from .utils import CocoDetection
|
||||
from .utils import COCOMetric
|
||||
import cv2
|
||||
from tqdm import trange
|
||||
|
||||
if conf_threshold is not None or nms_iou_threshold is not None:
|
||||
assert conf_threshold is not None and nms_iou_threshold is not None, "The conf_threshold and nms_iou_threshold should be setted at the same time"
|
||||
assert isinstance(conf_threshold, (
|
||||
|
Reference in New Issue
Block a user