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

Explore extreme compression for pre-trained language models

Parris, the automated infrastructure setup tool for machine learning algorithms.

[ICLR 2021] Heteroskedastic and Imbalanced Deep Learning with Adaptive Regularization

SOTA model in CIFAR10

Learning Domain Invariant Representations in Goal-conditioned Block MDPs

Conditional Gradients For The Approximately Vanishing Ideal

SEJE Pytorch implementation

Tensorflow 2 implementations of the C-SimCLR and C-BYOL self-supervised visual representation methods from "Compressive Visual Representations" (NeurIPS 2021)

A Closer Look at Reference Learning for Fourier Phase Retrieval

CTC segmentation python package

Deep Learning Head Pose Estimation using PyTorch.

Code for the Shortformer model, from the paper by Ofir Press, Noah A. Smith and Mike Lewis.

Dense Passage Retriever - is a set of tools and models for open domain Q&A task.

StyleMapGAN - Official PyTorch Implementation

An Open-Source Package for Information Retrieval.

ROS-UGV-Control-Interface - Control interface which can be used in any UGV

Pytorch implementation of MLP-Mixer with loading pre-trained models.

Code for "The Box Size Confidence Bias Harms Your Object Detector"

Pytorch implementation of AngularGrad: A New Optimization Technique for Angular Convergence of Convolutional Neural Networks

GE2340 project source code without credentials.