What dataset are these models pretrained on?

intel / ai-reference-models

Intel® AI Reference Models: contains Intel optimizations for running deep learning workloads on Intel® Xeon® Scalable processors and Intel® Data Center GPUs

Apache License 2.0

682 stars 219 forks source link

What dataset are these models pretrained on? #70

Open paulgavrikov opened 3 years ago

paulgavrikov commented 3 years ago

Could you please tell me what dataset the following models are pretrained on?

SSD-MobileNet (dataset combinedNMS?) https://github.com/IntelAI/models/tree/master/benchmarks/object_detection/tensorflow/ssd-mobilenet
SSD-ResNet34 https://github.com/IntelAI/models/tree/master/benchmarks/object_detection/tensorflow/ssd-resnet34
UNet https://github.com/IntelAI/models/tree/master/benchmarks/image_segmentation/tensorflow/unet

sramakintel commented 7 months ago

@paulgavrikov: UNet uses randomly generated images:https://github.com/jakeret/tf_unet/blob/master/tf_unet/image_gen.py SSD-ResNet34 and SSD-Mobilenet use COCO validation dataset in the TF records format. Let us know if this resolves your issue.