Skip to content

HiAliens/Cat-and-dog

Repository files navigation

Cat-and-dog

            使用AlexNet识别猫和狗
            使用的是tensorflow框架,项目里包括了若干剪裁过的图像,完整数据集在:
            链接:https://pan.baidu.com/s/1m0i9hlqH0o_xXXJGH6Vn5Q 
            提取码:h2v7 

            图片结构应该是:
            images--
               train --
                         cats --
                            cat.001.jpg
                            ... ...
                      dogs --
                                  dog.001.jpg
                                  ... ...
             test --
                        cats --
                             cat.001.jpg
                            ... ...
                       dogs --
                            dog.001.jpg
                            ... ...
                 val --
                       cats --
                           cat.001.jpg
                           ... ...
                         dogs --
                                 dog.001.jpg
                                  ... ...
            请务必注意图片文件夹名称和图片名称格式,代码中用到了文件夹名称来提取所属类别

            这里提供两种数据读取的方法
            1. 将图片数据转化为TFRecord文件,之后再读取(见converttotfrecord.py & readfromtfrecord.py这两个文件)
            2. 读取文件位置,在需要读取图片时分批次读取(见getbatch.py文件)

About

使用AlexNet识别猫和狗

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages