fix some typo test=develop

revert-15296-async_double_buffered_py_reader
tink2123 6 years ago committed by dengkaipeng
parent 2b89f59055
commit 20200e126d

@ -482,7 +482,7 @@ def yolov3_loss(x,
raise TypeError("Attr anchor_mask of yolov3_loss must be list or tuple")
if not isinstance(class_num, int):
raise TypeError("Attr class_num of yolov3_loss must be an integer")
if not isinstance(class_num, int):
if not isinstance(use_label_smooth, int):
raise TypeError("Attr ues_label_smooth of yolov3 must be a bool value")
if not isinstance(ignore_thresh, float):
raise TypeError(

@ -76,7 +76,7 @@ def YOLOv3Loss(x, gtbox, gtlabel, gtscore, attrs):
class_num = attrs["class_num"]
ignore_thresh = attrs['ignore_thresh']
downsample = attrs['downsample']
#use_label_smooth = attrs['use_label_smooth']
use_label_smooth = attrs['use_label_smooth']
input_size = downsample * h
x = x.reshape((n, mask_num, 5 + class_num, h, w)).transpose((0, 1, 3, 4, 2))
loss = np.zeros((n)).astype('float32')

Loading…
Cancel
Save