-
Yolact_minimal:PyTorch中对YOLACT的最小实现
资源介绍
Yolact_minimal
Yolact的最小PyTorch实现 。 原始项目在。
此实现简化了原始代码,保留了主要功能并使网络易于理解。 此实现尚未更新为Yolact ++。
以双绞变压器为骨干的新版本已经在市场上。 TensorRT加速度也已经存在。 这两个分支即将合并。 请检查
网络结构。
环境环境
PyTorch> = 1.1 Python> = 3.6 n 对于CUDA 10.2,onnxruntime-gpu == 1.6.0 tensooardX 其他常见的软件包。
准备
构建cython-nms
python setup.py build_ext --inplace
下载COCO 2017数据集,在config.py 'res101_coco'中修改self.data_root 。
下载权重。
Yolact训练的重量。
骨干
盒图
面膜MAP
Google云