# Mask RCNN
Mask RCNN in TensorFlow
This repo attempts to reproduce this amazing work by Kaiming He et al. :
[Mask R-CNN](https://arxiv.org/abs/1703.06870)
## Requirements
- [Tensorflow (>= 1.0.0)](https://www.tensorflow.org/install/install_linux)
- [Numpy](https://github.com/numpy/numpy/blob/master/INSTALL.rst.txt)
- [COCO dataset](http://mscoco.org/dataset/#download)
- [Resnet50](http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz)
## How-to
1. Go to `./libs/datasets/pycocotools` and run `make`
2. Download [COCO](http://mscoco.org/dataset/#download) dataset, place it into `./data`, then run `python download_and_convert_data.py` to build tf-records. It takes a while.
3. Download pretrained resnet50 model, `wget http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz`, unzip it, place it into `./data/pretrained_models/`
4. Go to `./libs` and run `make`
5. run `python train/train.py` for training
6. There are certainly some bugs, please report them back, and let's solve them together.
没有合适的资源?快使用搜索试试~ 我知道了~
tensorflow-使用tensorflow实现的Mask-RCNN算法.zip
共82个文件
py:60个
pyx:7个
gitignore:2个
0 下载量 72 浏览量
2024-03-02
19:28:38
上传
评论
收藏 266KB ZIP 举报
温馨提示
tensorflow_使用tensorflow实现的Mask-RCNN算法
资源推荐
资源详情
资源评论
收起资源包目录
tensorflow_使用tensorflow实现的Mask-RCNN算法.zip (82个子文件)
tensorflow_使用tensorflow实现的Mask-RCNN算法
train
__init__.py 64B
train_utils.py 6KB
train.py 14KB
data
README.md 142B
download_and_convert_data.py 1KB
libs
__init__.py 0B
layers
__init__.py 754B
crop.py 3KB
mask.py 5KB
sample.py 8KB
wrapper.py 7KB
roi.py 7KB
assign.py 1KB
anchor.py 9KB
setup.py 6KB
Makefile 67B
nms
__init__.py 0B
gpu_nms.pyx 1KB
nms_kernel.cu 5KB
.gitignore 15B
gpu_nms.hpp 146B
cpu_nms.pyx 2KB
py_cpu_nms.py 1KB
configs
__init__.py 0B
config_v1.py 9KB
nets
__init__.py 0B
nets_factory.py 1KB
resnet_v1.py 13KB
pyramid_network.py 25KB
train_utils.py 7KB
resnet_utils.py 10KB
visualization
__init__.py 0B
pil_utils.py 4KB
summary_utils.py 1KB
datasets
__init__.py 0B
coco.py 5KB
dataset_factory.py 934B
pycocotools
__init__.py 21B
setup.py 723B
coco.py 18KB
Makefile 199B
mask.py 4KB
common
maskApi.c 8KB
gason.h 3KB
gason.cpp 9KB
maskApi.h 2KB
_mask.pyx 11KB
cocoeval.py 23KB
download_and_convert_coco.py 18KB
boxes
blob.py 2KB
__init__.py 508B
cython_bbox_transform.py 398B
cython_bbox_transform.pyx 4KB
nms.pyx 4KB
cython_nms.py 387B
bbox.pyx 3KB
cython_anchor.py 390B
profile 169KB
gprof2dot.py 103KB
cython_bbox.py 388B
nms_wrapper.py 2KB
profile.png 110KB
timer.py 948B
roi.py 484B
bbox_transform.py 3KB
.gitignore 15B
cython_anchor.pyx 1KB
anchor.py 4KB
nms.py 1008B
logs
__init__.py 0B
log.py 398B
preprocessings
utils.py 6KB
__init__.py 0B
coco_v1.py 4KB
make.sh 75B
memory_util.py 13KB
unit_test
__init__.py 0B
layer_test.py 13KB
data_test.py 2KB
resnet50_test.py 8KB
preprocessing_test.py 2KB
README.md 1KB
共 82 条
- 1
资源评论
__AtYou__
- 粉丝: 1183
- 资源: 261
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- STM8L101F3P6单片机+CC1100模块433M遥控器设计硬件(原理图+PCB)工程文件.zip
- 上传下载铁人下载系统 Liuxing 1.0-liuxing1.0.rar
- 南京邮电大学数学实验实力雄厚,凭借其优秀的师资力量、丰富的实践教学资源和卓越的科研成果,成为国内一流的数学实验教学和科研基地
- 【火爆朋友圈的今天吃什么源码 v1.0】随机的为用户带来每一天的用餐选择和推荐.rar
- MPU6050中文版数据手册
- 上传下载手机电影下载-mobiledy.rar
- 响应式旅游网站源码下载 马尔代夫旅游网站.rar
- CMS小涴熊漫画连载系统漫画网站源码 带采集API.rar
- 福袋点点.apk
- 基于STM32的电子秤采用0.96寸OLED显示UI界面源码.zip
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功