Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-09-16 23:15:07 +02:00
parent 137aab762a
commit 425ed4b84b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models.py
View File

@ -1,8 +1,8 @@
import torch.nn.functional as F import torch.nn.functional as F
from utils.google_utils import *
from utils.parse_config import * from utils.parse_config import *
from utils.utils import * from utils.utils import *
from utils.google_utils import *
ONNX_EXPORT = False ONNX_EXPORT = False
@ -78,7 +78,7 @@ def create_modules(module_defs, img_size, arc):
# Initialize preceding Conv2d() bias (https://arxiv.org/pdf/1708.02002.pdf section 3.3) # Initialize preceding Conv2d() bias (https://arxiv.org/pdf/1708.02002.pdf section 3.3)
try: try:
if arc == 'defaultpw': # default with positive weights if arc == 'defaultpw' or arc == 'Fdefaultpw': # default with positive weights
b = [-4, -3.6] # obj, cls b = [-4, -3.6] # obj, cls
elif arc == 'default': # default no pw (40 cls, 80 obj) elif arc == 'default': # default no pw (40 cls, 80 obj)
b = [-5.5, -4.0] b = [-5.5, -4.0]