You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
onmywei 946ecd3f5b Fixed an coco evaluate error while training yourself dataset 3 years ago
..
tv_reference Fixed an coco evaluate error while training yourself dataset 3 years ago
__init__.py 1. demo and demo_onnx are broken 2. demo_onnx can do both converion and inference 3 years ago
camera.py Split utils into utils with torch and without torch 3 years ago
coco_annotation.py append GIoU DIoU CIoU,and models can be generated from *.cfg files during training 3 years ago
config.py Split utils into utils with torch and without torch 3 years ago
darknet2onnx.py Add support of dynamic batch size 3 years ago
darknet2pytorch.py Add support of dynamic batch size 3 years ago
onnx2tensorflow.py add onnx -> tensorflow 3 years ago
region_loss.py fix: deprecated 'size_average' 3 years ago
torch_utils.py Inference output as 2 tensors: [b, n, 1, 4] and [b, n, cls] 3 years ago
utils.py 1. classification should be sigmoid 2. NMS for each class 3 years ago
utils_iou.py replace the custom cosine similarity by torch.nn.functional.cosine_similarity, which is much faster 3 years ago
utils_iou_test.py replace the custom cosine similarity by torch.nn.functional.cosine_similarity, which is much faster 3 years ago
yolo_layer.py Add support of dynamic batch size 3 years ago