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