Fix requirements (#59)

* Fix bug in ppyoloe

* fix ppyoloe output format

* remove some requirements

* fix conflicts
This commit is contained in:
Jason
2022-07-31 15:05:30 +08:00
committed by GitHub
parent 4612159f99
commit ed3d6f2187
3 changed files with 9 additions and 11 deletions

View File

@@ -12,10 +12,8 @@
# See the License for the specific language governing permissions and
# limitations under the License.
import numpy as np
import cv2
import os
import re
from tqdm import trange
def topk_accuracy(topk_list, label_list):
@@ -25,6 +23,9 @@ def topk_accuracy(topk_list, label_list):
def eval_classify(model, image_file_path, label_file_path, topk=5):
from tqdm import trange
import cv2
result_list = []
label_list = []
image_label_dict = {}

View File

@@ -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, (

View File

@@ -1,5 +1 @@
opencv-python
tqdm
numpy
pycocotools
colorama
requests