Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details

All Projects → DetectionTeamUCAS → Fpn_tensorflow

DetectionTeamUCAS / Fpn_tensorflow

Licence: mit

This is a tensorflow re-implementation of Feature Pyramid Networks for Object Detection.

Labels

jupyter-notebook tensorflow object-detection

Projects that are alternatives of or similar to Fpn tensorflow

Shape Detection

🟣 Object detection of abstract shapes with neural networks

Stars: ✭ 170 (-41.58%)

Mutual labels: object-detection, jupyter-notebook

Deep Learning With Python

Deep learning codes and projects using Python

Stars: ✭ 195 (-32.99%)

Mutual labels: object-detection, jupyter-notebook

YoloV3 Implemented in Tensorflow 2.0

Stars: ✭ 2,327 (+699.66%)

Mutual labels: object-detection, jupyter-notebook

Ios Coreml Yolo

Almost Real-time Object Detection using Apple's CoreML and YOLO v1 -

Stars: ✭ 153 (-47.42%)

Mutual labels: object-detection, jupyter-notebook

🌮 Trash Annotations in Context Dataset Toolkit

Stars: ✭ 243 (-16.49%)

Mutual labels: object-detection, jupyter-notebook

Yolov4 Cloud Tutorial

This repository walks you through how to Build and Run YOLOv4 Object Detections with Darknet in the Cloud with Google Colab.

Stars: ✭ 153 (-47.42%)

Mutual labels: object-detection, jupyter-notebook

Py R Fcn Multigpu

Code for training py-faster-rcnn and py-R-FCN on multiple GPUs in caffe

Stars: ✭ 192 (-34.02%)

Mutual labels: object-detection, jupyter-notebook

Copy-paste augmentation for segmentation and detection tasks

Stars: ✭ 132 (-54.64%)

Mutual labels: object-detection, jupyter-notebook

Self hosted alternative to Google Photos

Stars: ✭ 2,587 (+789%)

Mutual labels: object-detection, jupyter-notebook

Traffic Sign Detection

Traffic Sign Detection. Code for the paper entitled "Evaluation of deep neural networks for traffic sign detection systems".

Stars: ✭ 200 (-31.27%)

Mutual labels: object-detection, jupyter-notebook

简明 SSD 目标检测模型 keras version（交通标志识别训练部分见 dev 分支）

Stars: ✭ 152 (-47.77%)

Mutual labels: object-detection, jupyter-notebook

Siamese Mask Rcnn

Siamese Mask R-CNN model for one-shot instance segmentation

Stars: ✭ 257 (-11.68%)

Mutual labels: object-detection, jupyter-notebook

Lacmus is a cross-platform application that helps to find people who are lost in the forest using computer vision and neural networks.

Stars: ✭ 142 (-51.2%)

Mutual labels: object-detection, jupyter-notebook

Face mask detection

Face mask detection system using Deep learning.

Stars: ✭ 168 (-42.27%)

Mutual labels: object-detection, jupyter-notebook

How to create custom COCO data set for object detection

Stars: ✭ 140 (-51.89%)

Mutual labels: object-detection, jupyter-notebook

FPGA-based neural network inference project with an end-to-end approach (from training to implementation to deployment)

Stars: ✭ 186 (-36.08%)

Mutual labels: object-detection, jupyter-notebook

Robust Detection Benchmark

Code, data and benchmark from the paper "Benchmarking Robustness in Object Detection: Autonomous Driving when Winter is Coming" (NeurIPS 2019 ML4AD)

Stars: ✭ 128 (-56.01%)

Mutual labels: object-detection, jupyter-notebook

Object detection demo

How to train an object detection model easy for free

Stars: ✭ 130 (-55.33%)

Mutual labels: object-detection, jupyter-notebook

Nas fpn tensorflow

NAS-FPN: Learning Scalable Feature Pyramid Architecture for Object Detection.

Stars: ✭ 198 (-31.96%)

Mutual labels: object-detection, jupyter-notebook

SipMask: Spatial Information Preservation for Fast Image and Video Instance Segmentation (ECCV2020)

Stars: ✭ 255 (-12.37%)

Mutual labels: object-detection, jupyter-notebook

View All Similar Projects ➔

Feature Pyramid Networks for Object Detection

Note

A development version based on FPN.
Support multi-gpu training!

Abstract

This is a tensorflow re-implementation of Feature Pyramid Networks for Object Detection.

This project is based on Faster-RCNN, and completed by YangXue and YangJirui.

Train on VOC 2007 trainval and test on VOC 2007 test (PS. This project also support coco training.)

Comparison

use_voc2007_metric

Models	mAP	sheep	horse	bicycle	bottle	cow	sofa	bus	dog	cat	person	train	diningtable	aeroplane	car	pottedplant	tvmonitor	chair	bird	boat	motorbike
Faster-RCNN resnet50_v1	73.09	72.11	85.63	77.74	55.82	81.19	67.34	82.44	85.66	87.34	77.49	79.13	62.65	76.54	84.01	47.90	74.13	50.09	76.81	60.34	77.47
Faster-RCNN resnet101_v1	74.63	76.35	86.18	79.87	58.73	83.4	74.75	80.03	85.4	86.55	78.24	76.07	70.89	78.52	86.26	47.80	76.34	52.14	78.06	58.90	78.04
Faster-RCNN mobilenet_v2	50.34	46.99	68.45	65.89	28.16	53.21	46.96	57.80	38.60	44.12	66.20	60.49	52.40	56.06	72.68	26.91	49.99	30.18	39.38	38.54	64.74
FPN resnet50_v1	74.26	73.27	82.23	82.99	61.27	80.59	72.73	81.37	85.26	84.76	80.33	77.43	65.31	79.18	85.78	46.47	73.10	55.99	76.11	59.80	81.19
FPN resnet101_v1	76.14	74.63	85.13	81.67	63.79	82.43	77.83	83.07	86.45	85.82	81.08	81.01	71.22	80.01	86.30	48.05	73.89	56.99	78.33	62.91	82.24
FPN resnet101_v1+	75.71	74.83	83.55	82.47	65.49	77.85	71.74	80.98	86.61	87.14	81.02	77.76	71.26	79.82	86.78	51.64	77.45	56.12	79.44	60.55	81.69
FPN resnet101_v1++	75.89	76.05	84.22	80.29	63.21	83.04	78.69	81.81	86.61	85.61	79.75	79.78	71.27	80.33	86.24	49.03	76.81	56.32	78.51	60.37	79.91

+: SHARE_NET=False
++: SHORT_SIDE_LEN=800, FAST_RCNN_MINIBATCH_SIZE=512

COCO

Model	Backbone	Train Schedule	GPU	Image/GPU	FP16	Box AP(Mask AP)
Faster (ours)	R50v1-FPN	1X	1X TITAN Xp	1	no	36.1
Faster (ours)	R50v1-FPN	1X	4X TITAN Xp	1	no	36.1
Faster (Face++ & Detectron)	R50v1-FPN	1X	8X TITAN Xp	2	no	36.4
Faster (SimpleDet)	R50v1-FPN	1X	8X 1080Ti	2	no	36.5

My Development Environment

1、python3.5 (anaconda recommend)
2、cuda9.0 (If you want to use cuda8, please set CUDA9 = False in the cfgs.py file.)
3、opencv(cv2)
4、tfplot
5、tensorflow == 1.10

Download Model

Please download resnet50_v1、resnet101_v1 pre-trained models on Imagenet, put it to $PATH_ROOT/data/pretrained_weights.

Data Format

├── VOCdevkit
│   ├── VOCdevkit_train
│       ├── Annotation
│       ├── JPEGImages
│   ├── VOCdevkit_test
│       ├── Annotation
│       ├── JPEGImages

Compile

cd $PATH_ROOT/libs/box_utils/cython_utils
python setup.py build_ext --inplace

Demo(available)

Select a configuration file in the folder ($PATH_ROOT/libs/configs/) and copy its contents into cfgs.py, then download the corresponding weights.

cd $PATH_ROOT/tools
python inference.py --data_dir='/PATH/TO/IMAGES/' 
                    --save_dir='/PATH/TO/SAVE/RESULTS/' 
                    --GPU='0'

Eval

cd $PATH_ROOT/tools
python eval.py --eval_imgs='/PATH/TO/IMAGES/'  
               --annotation_dir='/PATH/TO/TEST/ANNOTATION/'
               --GPU='0'

Train

1、If you want to train your own data, please note:

(1) Modify parameters (such as CLASS_NUM, DATASET_NAME, VERSION, etc.) in $PATH_ROOT/libs/configs/cfgs.py
(2) Add category information in $PATH_ROOT/libs/label_name_dict/lable_dict.py     
(3) Add data_name to line 76 of $PATH_ROOT/data/io/read_tfrecord.py

2、make tfrecord

cd $PATH_ROOT/data/io/  
python convert_data_to_tfrecord.py --VOC_dir='/PATH/TO/VOCdevkit/VOCdevkit_train/' 
                                   --xml_dir='Annotation'
                                   --image_dir='JPEGImages'
                                   --save_name='train' 
                                   --img_format='.jpg' 
                                   --dataset='pascal'

3、train

cd $PATH_ROOT/tools
python train.py

4、multi-gpu train

cd $PATH_ROOT/tools
python multi_gpu_train.py

Tensorboard

cd $PATH_ROOT/output/summary
tensorboard --logdir=.

Reference

1、https://github.com/endernewton/tf-faster-rcnn
2、https://github.com/zengarden/light_head_rcnn
3、https://github.com/tensorflow/models/tree/master/research/object_detection
4、https://github.com/CharlesShang/FastMaskRCNN
5、https://github.com/matterport/Mask_RCNN
6、https://github.com/msracver/Deformable-ConvNets

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].

Stars: ✭ 291

Visit Git Page 🔗Visit User Page 🔗Visit Issues Page (20) 🔗