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

RLDS stands for Reinforcement Learning Datasets

Deep Learning tutorials in jupyter notebooks.

Dimension Reduced Turbulent Flow Data From Deep Vector Quantizers

Pytorch implementation of ICASSP 2022 paper Attention Probe: Vision Transformer Distillation in the Wild

PyTorch implementation of Munchausen Reinforcement Learning based on DQN and SAC. Handles discrete and continuous action spaces

A collection of Google research projects related to Federated Learning and Federated Analytics.

A state of the art of new lightweight YOLO model implemented by TensorFlow 2.

PyTorch Implementation for Fracture Detection in Wrist Bone X-ray Images

Expressive Power of Invariant and Equivaraint Graph Neural Networks (ICLR 2021)

A series of Jupyter notebooks with Chinese comment that walk you through the fundamentals of Machine Learning and Deep Learning in python using Scikit-Learn and TensorFlow.

gtfs2vec - Learning GTFS Embeddings for comparing PublicTransport Offer in Microregions

[ICCV 2021] Amplitude-Phase Recombination: Rethinking Robustness of Convolutional Neural Networks in Frequency Domain

Pi-NAS: Improving Neural Architecture Search by Reducing Supernet Training Consistency Shift (ICCV 2021)

Code for "LoFTR: Detector-Free Local Feature Matching with Transformers", CVPR 2021

The Few-Shot Bot: Prompt-Based Learning for Dialogue Systems

An Unpaired Sketch-to-Photo Translation Model

Negative Sample is Negative in Its Own Way: Tailoring Negative Sentences forImage-Text Retrieval

Tensor-based approaches for fMRI classification

Main Results on ImageNet with Pretrained Models

Deduplicating Training Data Makes Language Models Better