LF-YOLO (Lighter and Faster YOLO) is used to detect defect of X-ray weld image.

Last update: Dec 13, 2022

Related tags

Deep Learning LF-YOLO

Overview

This project is based on ultralytics/yolov3.

LF-YOLO (Lighter and Faster YOLO) is used to detect defect of X-ray weld image.

Download

$ git clone https://github.com/lmomoy/LF-YOLO

Train

We provide multiple versions of LF-YOLO with different depths.

$ python train.py --data coco.yaml --cfg LF-YOLO.yaml      --weights '' --batch-size 1
                                         LF-YOLO-1.25.yaml                           1
                                         LF-YOLO-0.75.yaml                           1
                                         LF-YOLO-0.5.yaml                            1

Results

We test our model on public dataset MS COCO, and it also achieves competitive results.

Model	size ^(pixels)	mAP^val 0.5:0.95	mAP^{test 0.5:0.95}	mAP^val 0.5	params ^(M)	FLOPS ^{640 (B)}
YOLOv3-tiny	640	17.6	17.6	34.8	8.8	13.2
YOLOv3	640	43.3	43.3	63.0	61.9	156.3
YOLOv3-SPP	640	44.3	44.3	64.6	63.0	157.1
LF-YOLO	640	27.8	27.9	47.8	7.4	17.1

Table Notes (click to expand)

AP^test denotes COCO test-dev2017 server results, all other AP results denote val2017 accuracy.
AP values are for single-model single-scale unless otherwise noted. Reproduce mAP by python test.py --data coco.yaml --img 640 --conf 0.001 --iou 0.65
Speed_GPU averaged over 5000 COCO val2017 images using a GCP n1-standard-16 V100 instance, and includes FP16 inference, postprocessing and NMS. Reproduce speed by python test.py --data coco.yaml --img 640 --conf 0.25 --iou 0.45
All checkpoints are trained to 300 epochs with default settings and hyperparameters (no autoaugmentation).

Requirements

Python 3.8 or later with all requirements.txt dependencies installed, including torch>=1.7. To install run:

$ pip install -r requirements.txt

Inference

$ python detect.py --source data/images --weights LF-YOLO.pt --conf 0.25

LF-YOLO (Lighter and Faster YOLO) is used to detect defect of X-ray weld image.

Related tags

Overview

Download

Train

Results

Requirements

Inference

Owner

Alias-Free Generative Adversarial Networks (StyleGAN3) Official PyTorch implementation

Human head pose estimation using Keras over TensorFlow.

Self-Supervised Image Denoising via Iterative Data Refinement

NudeNet: Neural Nets for Nudity Classification, Detection and selective censoring

A Dataset for Direct Quotation Extraction and Attribution in News Articles.

PyTorch implementations of the paper: "DR.VIC: Decomposition and Reasoning for Video Individual Counting, CVPR, 2022"

TLDR: Twin Learning for Dimensionality Reduction

Toolchain to build Yoshi's Island from source code

AgML is a comprehensive library for agricultural machine learning

Developed an optimized algorithm which finds the most optimal path between 2 points in a 3D Maze using various AI search techniques like BFS, DFS, UCS, Greedy BFS and A*

KGDet: Keypoint-Guided Fashion Detection (AAAI 2021)

PyTea: PyTorch Tensor shape error analyzer

Prototypical Pseudo Label Denoising and Target Structure Learning for Domain Adaptive Semantic Segmentation (CVPR 2021)

End-to-End Referring Video Object Segmentation with Multimodal Transformers

Import Python modules from dicts and JSON formatted documents.

Reference implementation of code generation projects from Facebook AI Research. General toolkit to apply machine learning to code, from dataset creation to model training and evaluation. Comes with pretrained models.

Fast, flexible and fun neural networks.

Style-based Point Generator with Adversarial Rendering for Point Cloud Completion (CVPR 2021)

MT-GAN-PyTorch - PyTorch Implementation of Learning to Transfer: Unsupervised Domain Translation via Meta-Learning

Mae segmentation - Reproduction of semantic segmentation using masked autoencoder (mae)