Iranian Cars Detection using Yolov5

Train

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -r requirements.txt

2- Dataset

../Dataset/images/im0.jpg  # image
../Dataset/labels/im0.txt  # label

3- Create dataset.yaml

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]

path: ../Dataset  # dataset root dir
train: ../Dataset/Train/images  # train images (relative to 'path') 128 images
val: ../Dataset/Val/images  # val images (relative to 'path') 128 images

# Classes
nc: 5  # number of classes
names: ['iranKhodro_dena', 'kia_cerato', 'mazda_3', 'peugeot_206', 'saipa_saina']  # class names

You can change this sections in Yolov5/data/coco128.yaml.

4- Train YOLOv5s on Dataset for 30 epochs using following command:

$ python train.py --img 640 --batch 16 --epochs 30 --data coco128.yaml --weights yolov5s.pt

Inference

!python /yolov5/detect.py --weights /runs/train/exp13/weights/last.pt --img 244 --conf 0.4 --source /data/saipa_saina88.jpg

Iranian Cars Detection using Yolov5s, PyTorch

Related tags

Overview

Iranian Cars Detection using Yolov5

Train

Inference

Owner

Nahid Ebrahimian

PyTorch implementation of the method described in the paper VoiceLoop: Voice Fitting and Synthesis via a Phonological Loop.

PyTorch implementation of our ICCV paper DeFRCN: Decoupled Faster R-CNN for Few-Shot Object Detection.

3D Multi-Person Pose Estimation by Integrating Top-Down and Bottom-Up Networks

A little software to generate and save Julia or Mandelbrot's Fractals.

Codebase for Amodal Segmentation through Out-of-Task andOut-of-Distribution Generalization with a Bayesian Model

Walk with fastai

State of the Art Neural Networks for Generative Deep Learning

Distributing Deep Learning Hyperparameter Tuning for 3D Medical Image Segmentation

Selecting Parallel In-domain Sentences for Neural Machine Translation Using Monolingual Texts

Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...

Collective Multi-type Entity Alignment Between Knowledge Graphs (WWW'20)

Semi-Supervised Semantic Segmentation via Adaptive Equalization Learning, NeurIPS 2021 (Spotlight)

Code for CVPR 2018 paper --- Texture Mapping for 3D Reconstruction with RGB-D Sensor

Lex Rosetta: Transfer of Predictive Models Across Languages, Jurisdictions, and Legal Domains

Easily benchmark PyTorch model FLOPs, latency, throughput, max allocated memory and energy consumption

SatelliteNeRF - PyTorch-based Neural Radiance Fields adapted to satellite domain

Neural Architecture Search Powered by Swarm Intelligence 🐜

UCSD Oasis platform

Detection of drones using their thermal signatures from thermal camera through YOLO-V3 based CNN with modifications to encapsulate drone motion

A Benchmark For Measuring Systematic Generalization of Multi-Hierarchical Reasoning