PyTorch对象检测训练套件——从图像到视频流的全方位实现
本套件包括train_lenet.py,classify_image.py,detect_image.py,detect_realtime.py和predict.py等文件,同时还提供coco_classes.json,coco_classes.pickle和coco_classes_91.pickle等文件以帮助用户进行对象检测。此外,还附带了img_merge.py,coco_download.py和read_write_pickle.py等实用工具,方便用户的操作。
下载地址
用户评论