Update check_requirements() exclude list (#2974)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-04-29 21:16:23 +02:00
1 parent dbce1bc54c
commit a833ee2a46
6 files changed
+8 -6

No files matched your search

+2 -1
View File
@@ -3,7 +3,6 @@
import numpy as np
import torch
import yaml
from scipy.cluster.vq import kmeans
from tqdm import tqdm
from utils.general import colorstr
@@ -76,6 +75,8 @@ def kmean_anchors(path='./data/coco128.yaml', n=9, img_size=640, thr=4.0, gen=10
Usage:
from utils.autoanchor import *; _ = kmean_anchors()
"""
from scipy.cluster.vq import kmeans
thr = 1. / thr
prefix = colorstr('autoanchor: ')