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