CA⫶TR :带有变压器的图像字幕
针对CATR的PyTorch培训代码和预训练模型( CA ption TR ansformer)。
也可以通过割炬轮毂使用这些模型,以简单地加载具有预训练砝码的模型:
model = torch . hub . load ( 'saahiluppal/catr' , 'v3' , pretrained = True ) # you can choose between v1, v2 and v3
样品:
所有这些图像均已由CATR注释。
使用您自己的图像进行测试:
$ python predict.py --path /path/to/image --v v2 // You can choose between v1, v2, v3 [default is v3]
或在colab中试用
用法
CATR中没有多余的已编译组件,并且程序包依赖
1