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

Empirical Study of Transformers for Source Code & A Simple Approach for Handling Out-of-Vocabulary Identifiers in Deep Learning for Source Code

This repository contains python code necessary to replicated the experiments performed in our paper "Invariant Ancestry Search"

This repository contains code, network definitions and pre-trained models for working on remote sensing images using deep learning

The repo of the preprinting paper "Labels Are Not Perfect: Inferring Spatial Uncertainty in Object Detection"

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

Perform Linear Classification with Multi-way Data

The codes of paper 'Active-LATHE: An Active Learning Algorithm for Boosting the Error exponent for Learning Homogeneous Ising Trees'

RP-GAN: Stable GAN Training with Random Projections

Streamlit App For Product Analysis - Streamlit App For Product Analysis

NitroFE is a Python feature engineering engine which provides a variety of modules designed to internally save past dependent values for providing continuous calculation.

ppo_pytorch_cpp - an implementation of the proximal policy optimization algorithm for the C++ API of Pytorch

RM Operation can equivalently convert ResNet to VGG, which is better for pruning; and can help RepVGG perform better when the depth is large.

Out-of-distribution detection using the pNML regret. NeurIPS2021

StarGAN - Official PyTorch Implementation (CVPR 2018)

PoseCamera is python based SDK for human pose estimation through RGB webcam.

Official PyTorch code for Hierarchical Conditional Flow: A Unified Framework for Image Super-Resolution and Image Rescaling (HCFlow, ICCV2021)

Event-forecasting - Event Forecasting Algorithms With Python

[ICCV'21] PlaneTR: Structure-Guided Transformers for 3D Plane Recovery

Code for ViTAS_Vision Transformer Architecture Search

Cryptocurrency Prediction with Artificial Intelligence (Deep Learning via LSTM Neural Networks)