Skip to content

syshensyshen/pva-faster-rcnn-pytorch-1.0

Repository files navigation

functions: pva-pytorch using pytorch-1.0 add multi-scales image train

pytorch1.0,opencv-python,skimage

step1: install git clone https://github.com/syshensyshen/pva-pytorch.git

'cd pva-pytorch/lib' python setup.py build develop

step2: train CUDA_VISIBLE_DEVICES=1 python trainval_net.py --batch_size 8
--save_dir ./save_models
--xml_path /data/datasets/defect_detection/voc/VOC2007/Annotations/
--img_path /data/datasets/defect_detection/voc/VOC2007/JPEGImages/
--network pva --classes 6

pretrainedmode:链接:https://pan.baidu.com/s/1eLdUheV3FHmsyYs-MQLKBQ 提取码:psnw

reference code: jwyang/faster-rcnn.pytorch sanghoon/pytorch_imagenet

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages