Files

ipex

yolov7 detection model

Jun 26, 2023

7cefdd1 · Jun 26, 2023

This branch is 54 commits ahead of hack2skill/intel-oneAPI:main.

Name		Name	Last commit message	Last commit date
parent directory ..
cfg/training		cfg/training	training module	Jun 8, 2023
data		data	training module	Jun 8, 2023
deploy/triton-inference-server		deploy/triton-inference-server	training module	Jun 8, 2023
models		models	training module	Jun 8, 2023
runs/train/yolov7-custom7		runs/train/yolov7-custom7	training module	Jun 8, 2023
scripts		scripts	training module	Jun 8, 2023
seg		seg	instance segmentation folder	Jun 26, 2023
utils		utils	training module	Jun 8, 2023
detect.py		detect.py	training module	Jun 8, 2023
export.py		export.py	training module	Jun 8, 2023
hubconf.py		hubconf.py	training module	Jun 8, 2023
readme.md		readme.md	Create readme.md	Jun 14, 2023
test.py		test.py	training module	Jun 8, 2023
train.py		train.py	Update train.py	Jun 14, 2023
train_aux.py		train_aux.py	training module	Jun 8, 2023
yolov7-seg.pt		yolov7-seg.pt	segmentation inference model	Jun 26, 2023
yolov7.pt		yolov7.pt	yolov7 detection model	Jun 26, 2023

readme.md

Yolov7 Model Training with IPEX

Software/Hardware Requirements

Linux OS or Docker Image on Windows docker pull intel/intel-optimized-pytorch:latest
Install PyTorch CPU python -m pip install torch torchvision
Install IPEX (Same version as PyTorch) python -m pip install intel_extension_for_pytorch -f https://developer.intel.com/ipex-whl-stable-cpu

Training

python train.py --workers 8 --device cpu --batch-size 32 --epoch 10 --data data/custom_data.yaml --hyp data/hyp.scratch.custom.yaml --img 640 640 --cfg cfg/training/yolov7-custom.yaml --weights yolov7.pt --name yolov7-custom

Inference

python detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --source oneAPI_ODAV_APP/TEST_VIDEO/Visual_Pollution.mp4