mmcv 是一个基础库,主要分为两个部分,一部分是和 deep learning framework 无关的一些工具函数,比如 IO/Image/Video 相关的一些操作,另一部分是为 PyTorch 写的一套训练工具,可以大大减少用户需要写的代码量,同时让整个流程的定制变得容易。
https://github.com/open-mmlab/mmcvgithub.commmcv在安装过程中需要VC++支持,安装过程中出现error:Microsoft Visual C++ 14.0 is required 参照这篇文章解决
HUST小菜鸡:Microsoft Visual C++ 14.0 is required解决方法zhuanlan.zhihu.comFile IO
该模块提供两个通用的接口用于加载和转储不同形式的文件
Load and dump data
import
扩充接口以支持更多的文件形式,需要写一个文件句柄继承BaseFileHandler然后将其注册为其他的文件类型。
@mmcv.register_handler('txt')
class TxtHandler1(mmcv.BaseFileHandler):
def load_from_fileobj(self, file):
return file.read()
def dump_to_fileobj(self, obj, file):
file.wr