diff --git a/detect.py b/detect.py index e1487364..6ef3a00f 100644 --- a/detect.py +++ b/detect.py @@ -40,11 +40,12 @@ def detect(cfg, _ = load_darknet_weights(model, weights) # Fuse Conv2d + BatchNorm2d layers - model.fuse() + # model.fuse() # Eval mode model.to(device).eval() + # Export mode if ONNX_EXPORT: img = torch.zeros((1, 3, s[0], s[1])) torch.onnx.export(model, img, 'weights/export.onnx', verbose=True) diff --git a/requirements.txt b/requirements.txt index 14bae566..ee50f163 100755 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,7 @@ # conda install numpy opencv matplotlib tqdm pillow # conda install pytorch torchvision -c pytorch # conda install scikit-image -c conda-forge +# conda install -c spyder-ide spyder-line-profiler numpy opencv-python torch >= 1.1.0