开源软件名称(OpenSource Name):ckczzj/Image-Object-Localization开源软件地址(OpenSource Url):https://github.com/ckczzj/Image-Object-Localization开源编程语言(OpenSource Language):Python 100.0%开源软件介绍(OpenSource Introduction):DescriptionGiven a picture with a bird, we are supposed to box the bird. In src/data directory, Neural NetworkFor traditional CNN and FC, it will meet degeneration problems when layers go deep. In paper These blocks compose ResNet: I use ResNet-18 in this project by adding a 4-dimension layer after ResNet-18 to predict box's x, y ,w and h. Loss: smooth l1 loss Metric: IoU of groound truth and prediction, threshold=0.75 TrainResize all images to Then normalize and standardize all pixel channel. Split all data into 9000 training data and 2788 tesing data. Train network on training data using Training result: Testing result: ExamplesRed box represents ground truth and green box is the prediction of network. Failed example: AttentionYou should keep the directory structure. Dependencypython 3.6 tensorflow 1.3.0 keras 2.1.0 numpy PIL pickle matplotlib RunIn
If you want to train model, If you want to test on trained model(if you had trained model), ReferenceDeep Residual Learning for Image Recognition: https://arxiv.org/pdf/1512.03385.pdf Authorckczzj LicenceMIT |
2023-10-27
2022-08-15
2022-08-17
2022-09-23
2022-08-13
请发表评论