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

Machine Learning Privacy Meter: A tool to quantify the privacy risks of machine learning models with respect to inference attacks, notably membership inference attacks

Unofficial PyTorch Implementation of "Augmenting Convolutional networks with attention-based aggregation"

Tensorflow2.0 🍎🍊 is delicious, just eat it! 😋😋

Information Gain Filtration (IGF) is a method for filtering domain-specific data during language model finetuning. IGF shows significant improvements over baseline fine-tuning without data filtration.

OpenMMLab Model Deployment Toolset

PiCIE: Unsupervised Semantic Segmentation using Invariance and Equivariance in clustering (CVPR2021)

The code is for the paper "A Self-Distillation Embedded Supervised Affinity Attention Model for Few-Shot Segmentation"

Canonical Capsules: Unsupervised Capsules in Canonical Pose (NeurIPS 2021)

Classification Modeling: Probability of Default

🚩🚩🚩

Applications using the GTN library and code to reproduce experiments in "Differentiable Weighted Finite-State Transducers"

GeneralOCR is open source Optical Character Recognition based on PyTorch.

Implementation of the "PSTNet: Point Spatio-Temporal Convolution on Point Cloud Sequences" paper.

meProp: Sparsified Back Propagation for Accelerated Deep Learning

An implementation of Geoffrey Hinton's paper "How to represent part-whole hierarchies in a neural network" in Pytorch.

Creating a Linear Program Solver by Implementing the Simplex Method in Python with NumPy

Official repository for "Restormer: Efficient Transformer for High-Resolution Image Restoration". SOTA results for single-image motion deblurring, image deraining, image denoising (synthetic and real data), and dual-pixel defocus deblurring.

PyZebrascope - an open-source Python platform for brain-wide neural activity imaging in behaving zebrafish

Compare GAN code.

The official implementation of Variable-Length Piano Infilling (VLI).