updates

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)

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