Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Jay Meng 693a3af200 | 3 years ago | |
---|---|---|
.idea | 3 years ago | |
android | 3 years ago | |
core | 3 years ago | |
data | 3 years ago | |
font | 3 years ago | |
mAP | 3 years ago | |
scripts | 3 years ago | |
.gitignore | 3 years ago | |
README.md | 3 years ago | |
benchmarks.py | 3 years ago | |
convert_tflite.py | 3 years ago | |
convert_trt.py | 3 years ago | |
demo.py | 3 years ago | |
detect.py | 3 years ago | |
detectvideo.py | 3 years ago | |
evaluate.py | 3 years ago | |
freeze_ckpt_to_pb.py | 3 years ago | |
predict.py | 3 years ago | |
requirements-gpu.txt | 3 years ago | |
requirements.txt | 3 years ago | |
save_model.py | 3 years ago | |
show_layer_feature_map.py | 3 years ago | |
test.py | 3 years ago | |
train.py | 3 years ago |
A tensorflow implementation of YOLOv5 inspired by https://github.com/ultralytics/yolov5.
A tensorflow implementation of YOLOv4 inspired by https://github.com/AlexeyAB/darknet.
Frame code from https://github.com/YunYang1994/tensorflow-yolov3.
Backbone: Darknet53; CSPDarknet53[1], Mish[2]; MobileNetV2; MobileNetV3(large and small)
Head: YOLOv5/YOLOv4(Mish), YOLOv3(Leaky_ReLU)[10];
Loss: DIOU CIOU[5], Focal_Loss[6]; Other: Label_Smoothing[7];
Python 3.6.8
Tensorflow 1.13.1 or Tensorflow 2.0 up
Running from_darknet_weights_to_ckpt.py will get tf yolov4 weight file yolov4_coco.ckpt.
python scripts/from_darknet_weights_to_ckpt.py
Running ckpt2pb.py will get tf yolov4 weight file yolov4.pb.
python scripts/ckpt2pb.py
Or running from_darknet_weights_to_pb.py directly.
python scripts/from_darknet_weights_to_pb.py
In core/config.py add your own path.
usage: python train.py gpu_id net_type(yolov5/yolov4/yolov3/tiny)
python train.py 0 yolov5
Inference
python test.py
python demo.py
[1] Cross Stage Partial Network (CSPNet)
[2] A Self Regularized Non-Monotonic Neural Activation Function
[3] Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition
[4] Path Aggregation Network for Instance Segmentation
[5] Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression
[6] Focal Loss for Dense Object Detection
[7] When Does Label Smoothing Help?
[8] Convolutional Block Attention Module
[9] YOLOv4: Optimal Speed and Accuracy of Object Detection
[10] YOLOv3: An Incremental Improvement
[11] Aggregated Residual Transformations for Deep Neural Networks
keras_yolov3 https://github.com/qqwweee/keras-yolo3.
keras_yolov4 https://github.com/Ma-Dan/keras-yolo4.
No Description
Python Java Text Gradle
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》